diff --git a/models/models.py b/models/models.py index 085b530..afa0942 100644 --- a/models/models.py +++ b/models/models.py @@ -20,7 +20,7 @@ def __call__( # Use standard coordinate tensor if none is given if coordinate_tensor is None: - coordinate_tensor = self.make_coordinate_slice( + coordinate_tensor = general.make_coordinate_slice( output_shape, dimension, slice_pos )