From 06048448417b08a07669d060fcd5f3121602399e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Mateusz=20Sok=C3=B3=C5=82?= Date: Mon, 20 May 2024 17:20:23 +0200 Subject: [PATCH] Adjustments for Finch.jl#562 --- src/finch/julia.py | 2 +- src/finch/tensor.py | 6 +++++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/src/finch/julia.py b/src/finch/julia.py index bca1743..b53b7e4 100644 --- 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 index 3572ac5..ef781a8 100644 --- 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)