Missing reference handling

Quasars · Oct 1, 2024 · 4339f45 · 4339f45
1 parent bf4d520
commit 4339f45
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/orangecontrib/snom/widgets/preprocessors/self_reference.py b/orangecontrib/snom/widgets/preprocessors/self_reference.py
@@ -13,14 +13,17 @@
 
 class SelfRef(PreprocessImageOpts2DOnlyWhole):
     def __init__(self, reference):
-        self.reference = reference.X
+        self.reference = reference
 
     def transform_image(self, image):
-        ref = (np.reshape(self.reference,np.shape(image)))
-        d = SelfReference(referencedata=ref).transform(
-            image
-        )
-        return d
+        try:
+            ref = (np.reshape(self.reference.X,np.shape(image)))
+            d = SelfReference(referencedata=ref).transform(
+                image
+            )
+            return d
+        except:
+            return image
 
 
 class SelfRefEditor(BaseEditorOrange):