Skip to content

Commit bd96bf7

Browse files
committed
clean up [skip ci]
1 parent 8e0a7b6 commit bd96bf7

File tree

2 files changed

+2
-10
lines changed

2 files changed

+2
-10
lines changed

script/conda.sh

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -24,9 +24,3 @@ fi
2424
conda update --yes conda
2525

2626
conda create --yes -n test python="${PYTHON_VERSION}"
27-
28-
gcc --version
29-
30-
if [ "${TRAVIS_OS_NAME}" = "linux" ]; then
31-
conda install gcc_linux-64 --yes
32-
fi

torch_scatter/scatter.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -50,11 +50,9 @@ def scatter_mean(src: torch.Tensor, index: torch.Tensor, dim: int = -1,
5050
count.clamp_(1)
5151
count = broadcast(count, out, dim)
5252
if torch.is_floating_point(out):
53-
out.div_(count)
54-
# out.true_divide_(count)
53+
out.true_divide_(count)
5554
else:
56-
out.div_(count)
57-
# out.floor_divide_(count)
55+
out.floor_divide_(count)
5856
return out
5957

6058

0 commit comments

Comments
 (0)