remove op=False

ASEM000 · Aug 5, 2022 · dbcb683 · dbcb683
1 parent 1247e5a
commit dbcb683
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/kernex/src/map.py b/kernex/src/map.py
@@ -54,7 +54,7 @@ def __multi_call__(self, array, *args, **kwargs):
         return result.reshape(*self.output_shape, *func_shape)
 
 
-@pytc.treeclass(op=False)
+@pytc.treeclass
 class kernelMap(baseKernelMap):
     def __init__(self, func_dict, shape, kernel_size, strides, padding, relative):
 
@@ -64,7 +64,7 @@ def __call__(self, array, *args, **kwargs):
         return self.__call__(array, *args, **kwargs)
 
 
-@pytc.treeclass(op=False)
+@pytc.treeclass
 class offsetKernelMap(kernelMap):
     def __init__(self, func_dict, shape, kernel_size, strides, offset, relative):
 

diff --git a/kernex/src/scan.py b/kernex/src/scan.py
@@ -62,7 +62,7 @@ def scan_body(padded_array, view):
         )
 
 
-@pytc.treeclass(op=False)
+@pytc.treeclass
 class kernelScan(baseKernelScan):
     def __init__(self, func_dict, shape, kernel_size, strides, padding, relative):
 
@@ -72,7 +72,7 @@ def __call__(self, array, *args, **kwargs):
         return self.__call__(array, *args, **kwargs)
 
 
-@pytc.treeclass(op=False)
+@pytc.treeclass
 class offsetKernelScan(kernelScan):
     def __init__(self, func_dict, shape, kernel_size, strides, offset, relative):