clamp value for load sampler

paquiteau · Nov 21, 2024 · d6a6dad · d6a6dad
1 parent 8d9be7f
commit d6a6dad
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/snake/core/sampling/samplers.py b/src/snake/core/sampling/samplers.py
@@ -213,7 +213,8 @@ class LoadTrajectorySampler(NonCartesianAcquisitionSampler):
     def _single_frame(self, sim_conf: SimConfig) -> NDArray:
         """Load the trajectory."""
         data = read_trajectory(self.path, raster_time=self.raster_time)[0]
-        data /= 0.5 * data.max()
+        data = np.minimum(data, 0.5)
+        data = np.maximum(data, -0.5)
         return data