Adjustments for Finch.jl#562

finch-tensor · May 21, 2024 · 0604844 · 0604844
1 parent 2567a83
commit 0604844
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/src/finch/julia.py b/src/finch/julia.py
@@ -3,7 +3,7 @@
 import juliapkg
 
 _FINCH_NAME = "Finch"
-_FINCH_VERSION = "0.6.28"
+_FINCH_VERSION = "0.6.29"
 _FINCH_HASH = "9177782c-1635-4eb9-9bfb-d9dfa25e6bce"
 _FINCH_REPO_PATH = os.environ.get("FINCH_REPO_PATH", default=None)
 

diff --git a/src/finch/tensor.py b/src/finch/tensor.py
@@ -742,7 +742,11 @@ def _reduce(x: Tensor, fn: Callable, axis, dtype=None):
     else:
         result = fn(x._obj)
 
-    if jl.isa(result, jl.Finch.Tensor) or jl.isa(result, jl.Finch.LazyTensor):
+    if (
+        jl.isa(result, jl.Finch.SwizzleArray) or
+        jl.isa(result, jl.Finch.Tensor) or
+        jl.isa(result, jl.Finch.LazyTensor)
+    ):
         result = Tensor(result)
     else:
         result = np.array(result)