diff --git a/minerva/engines/patch_inferencer_engine.py b/minerva/engines/patch_inferencer_engine.py index 49a99ea..f9c050e 100644 --- a/minerva/engines/patch_inferencer_engine.py +++ b/minerva/engines/patch_inferencer_engine.py @@ -321,7 +321,7 @@ def __call__( tuple( [ slice(i, None) # TODO: if ((i + base >= 0) and (i < in_dim)) - for i, base, in_dim in zip([0, *offset], base, x.shape) + for i, in_dim in zip([0, *offset], x.shape) ] ) for offset in offsets