Rename impl

Quantco · Aug 1, 2024 · 6f47d5d · 6f47d5d
1 parent 1c0cac9
commit 6f47d5d
Show file tree

Hide file tree

Showing 7 changed files with 8 additions and 6 deletions.
diff --git a/ndonnx/_core/__init__.py b/ndonnx/_core/__init__.py
@@ -4,7 +4,7 @@
 from ._numericimpl import NumericOperationsImpl
 from ._stringimpl import StringOperationsImpl
 from ._boolimpl import BooleanOperationsImpl
-from ._interface import OperationsBlock
+from ._default import OperationsBlock
 
 __all__ = [
     "NumericOperationsImpl",

diff --git a/ndonnx/_core/_boolimpl.py b/ndonnx/_core/_boolimpl.py
@@ -13,7 +13,7 @@
 import ndonnx._data_types as dtypes
 import ndonnx._opset_extensions as opx
 
-from ._interface import OperationsBlock
+from ._default import OperationsBlock
 from ._utils import binary_op, unary_op
 
 if TYPE_CHECKING:

diff --git a/ndonnx/_core/_interface.py → ndonnx/_core/_default.py b/ndonnx/_core/_interface.py → ndonnx/_core/_default.py
diff --git a/ndonnx/_core/_numericimpl.py b/ndonnx/_core/_numericimpl.py
@@ -18,7 +18,7 @@
 import ndonnx._opset_extensions as opx
 from ndonnx._utility import promote
 
-from ._interface import OperationsBlock
+from ._default import OperationsBlock
 from ._utils import (
     binary_op,
     from_corearray,

diff --git a/ndonnx/_core/_stringimpl.py b/ndonnx/_core/_stringimpl.py
@@ -11,7 +11,7 @@
 import ndonnx._data_types as dtypes
 import ndonnx._opset_extensions as opx
 
-from ._interface import OperationsBlock
+from ._default import OperationsBlock
 from ._utils import binary_op
 
 if TYPE_CHECKING:

diff --git a/ndonnx/_data_types/structtype.py b/ndonnx/_data_types/structtype.py
@@ -8,7 +8,7 @@
 import numpy as np
 import typing_extensions
 
-from ndonnx._core._interface import OperationsBlock
+from ndonnx._core._default import OperationsBlock
 from ndonnx._data_types.coretype import CoreType
 
 from .schema import Schema

diff --git a/ndonnx/_funcs.py b/ndonnx/_funcs.py
@@ -675,7 +675,9 @@ def where(condition, x, y):
 
 # set.py
 def unique_all(x: Array):
-    return x.dtype._ops.unique_all(x)
+    if (out := x.dtype._ops.unique_all(x)) is not NotImplemented:
+        return out
+    raise UnsupportedOperationError(f"Unsupported operand type for unique: '{x.dtype}'")
 
 
 def unique_counts(x):