reset ai between runs

silx-kit · Jun 28, 2023 · fc0ad25 · fc0ad25
1 parent 0ded734
commit fc0ad25
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pyFAI/test/test_azimuthal_integrator.py b/pyFAI/test/test_azimuthal_integrator.py
@@ -661,7 +661,7 @@ def print_mem():
                 logger.error("psutil missing")
             else:
                 logger.warning("Memory consumption: %s",psutil.virtual_memory())
-        ai = AzimuthalIntegrator.sload(self.ai) #make an empty copy and work on just one module
+        ai = AzimuthalIntegrator.sload(self.ai) #make an empty copy and work on just one module of the detector (much faster)
         ai.detector = detector_factory("Pilatus_100k")
         img = self.img[:ai.detector.shape[0],:ai.detector.shape[1]]
 
@@ -676,6 +676,7 @@ def print_mem():
         failed = []
         for m in methods.values():
             res = ai.integrate2d_ng(img, 11, 13, variance=variance, error_model=error_model, method=m)
+            ai.reset()
             v = res.sum_variance
             if v.min()< 0:
                 failed.append(f"min variance is positive or null with {res.method}, error model {error_model.as_str()}")