add error message if detector is missing from factory

maxiv-science · Nov 8, 2023 · b38278b · b38278b
1 parent 8bb6ffc
commit b38278b
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/azint/detector.py b/azint/detector.py
@@ -1,3 +1,4 @@
+import inspect
 try:
     import h5py
 except ImportError:
@@ -23,7 +24,11 @@ def __init__(self, pixel1, pixel2, max_shape, pixel_corners=None):
 
     @classmethod
     def factory(cls, name, config = {}):
-        return globals()[name](**config)
+        cls = globals().get(name)
+        if inspect.isclass(cls) and issubclass(cls, Detector):
+            return cls(**config)
+        else:
+            raise RuntimeError(f'Detector {name} not supported')
 
 
     def get_pixel_corners(self, pixel1, pixel2, shape):