Binary special methods return NotImplemented (#1191)

* return NotImplemented on DNDarray methods * fix bug lshift * fix some bugs * add out/where params for binary functions * add reflective tests & out,where * test NotImplemented * add missing tests for truediv and matmul * fix test divmod * rearange checks divmod out * tests in divmod and sub * fix brackets in divmod test * revert changes *shift for older pytorch versions * add out, where parameters * update examples output from scalar inputs --------- Co-authored-by: Fabian Hoppe <[email protected]> Co-authored-by: Claudia Comito <[email protected]>
helmholtz-analytics · Nov 2, 2023 · f62e2b1 · f62e2b1 · github-actions · Nov 2, 2023
1 parent eb8ce2d
commit f62e2b1
Show file tree

Hide file tree

Showing 6 changed files with 714 additions and 114 deletions.
diff --git a/heat/core/_operations.py b/heat/core/_operations.py
@@ -24,7 +24,7 @@ def __binary_op(
     t1: Union[DNDarray, int, float],
     t2: Union[DNDarray, int, float],
     out: Optional[DNDarray] = None,
-    where: Optional[DNDarray] = True,
+    where: Union[bool, DNDarray] = True,
     fn_kwargs: Optional[Dict] = {},
 ) -> DNDarray:
     """