Merge pull request #42 from levje/levje/mps-support

Delay removing cache
scil-vital · May 29, 2024 · ddbaa2a · ddbaa2a
2 parents 66808b5 + bc9605b
commit ddbaa2a
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/TrackToLearn/environments/env.py b/TrackToLearn/environments/env.py
@@ -538,8 +538,7 @@ def _format_state(
         signal, _ = interpolate_volume_in_neighborhood(
             self.data_volume,
             coords,
-            self.neighborhood_directions,
-            clear_cache=False)
+            self.neighborhood_directions)
         N, S = signal.shape
 
         # Placeholder for the final imputs