Merge pull request #92 from neuro-ml/develop

faster Average
neuro-ml · Nov 25, 2024 · 4a94f0c · 4a94f0c
2 parents fcd06be + 6e8e13d
commit 4a94f0c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/dpipe/__version__.py b/dpipe/__version__.py
@@ -1 +1 @@
-__version__ = '0.4.2'
+__version__ = '0.4.3'
diff --git a/dpipe/im/grid.py b/dpipe/im/grid.py
@@ -142,7 +142,7 @@ def build(self):
             result_torch = torch.from_numpy(self._result)
             counts_torch = torch.from_numpy(self._counts)
 
-            counts_torch[counts_torch == 0] = 1
+            counts_torch.clip_(1)
             torch.div(result_torch, counts_torch, out=result_torch)
         else:
             np.true_divide(self._result, self._counts, out=self._result, where=self._counts > 0)