Skip to content

Commit

Permalink
bring #432 fix instead
Browse files Browse the repository at this point in the history
  • Loading branch information
dkazanc committed Sep 9, 2024
1 parent a1a826f commit 8dfcff4
Show file tree
Hide file tree
Showing 2 changed files with 8 additions and 5 deletions.
1 change: 1 addition & 0 deletions docs/source/conf.py
Original file line number Diff line number Diff line change
Expand Up @@ -52,6 +52,7 @@
for mod_name in MOCK_MODULES:
sys.modules[mod_name] = mock.Mock()


class CustomMock(mock.Mock):
def __repr__(self):
return "<cp.ndarray>"
Expand Down
12 changes: 7 additions & 5 deletions httomo/method_wrappers/dezinging.py
Original file line number Diff line number Diff line change
Expand Up @@ -48,11 +48,13 @@ def execute(self, block: T) -> T:
# check if data needs to be transfered host <-> device
block = self._transfer_data(block)

args = self._build_kwargs(self._transform_params(self._config_params), block)
# plug in the correct value for axis instead of auto. In case if axis is absent we take
# it equal to 0. This is to avoid failure, but (missing) template parameters should be
# tested in the yaml checker potentially
if self.package_name == "tomopy":
if "axis" in self.parameters:
args = self._build_kwargs(
self._transform_params(self._config_params), block
)
# plug in the correct value for axis instead of auto. In case if axis is absent we
# take it equal to 0. This is to avoid failure, but (missing) template parameters
# should be tested in the yaml checker potentially
self._config_params["axis"] = args.get("axis", 0)

with catch_gputime() as t:
Expand Down

0 comments on commit 8dfcff4

Please sign in to comment.