Code improvements for previous commit

mxcube · Oct 3, 2024 · 620121b · 620121b
1 parent 13f185e
commit 620121b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/mxcubecore/HardwareObjects/DESY/P11Collect.py b/mxcubecore/HardwareObjects/DESY/P11Collect.py
@@ -26,6 +26,7 @@
 import time
 import sys
 import os
+import math
 import logging
 import traceback
 import h5py
@@ -115,7 +116,7 @@ def set_resolution(self, value):
         Args:
             value (float): Resolution value to set.
         """
-        if round(HWR.beamline.resolution.get_value(), 2) != round(value, 2):
+        if math.isclose(HWR.beamline.resolution.get_value(), value, abs_tol=0.01):
             super().set_resolution(value)
 
     def data_collection_hook(self):

diff --git a/mxcubecore/HardwareObjects/abstract/AbstractCollect.py b/mxcubecore/HardwareObjects/abstract/AbstractCollect.py
@@ -223,11 +223,10 @@ def do_collect(self, owner):
             wavelength = self.current_dc_parameters.get("wavelength")
             energy = self.current_dc_parameters.get("energy")
             detector_distance = self.current_dc_parameters.get("detector_distance")
-            dd0 = self.current_dc_parameters.get("resolution")
-            if dd0 and dd0.get('upper'):
-                resolution = dd0["upper"]
-            else:
-                resolution = None
+            try:
+               resolution = self.current_dc_parameters.get("resolution").get("upper")
+            except AttributeError:
+               resolution = None
 
             if wavelength:
                 # Wavelength (not having a default) overrides energy