changing from ValueError to Exception

alex-rakowski · Nov 10, 2023 · 08957ff · 08957ff
1 parent be609b3
commit 08957ff
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/py4DSTEM/process/classification/featurization.py b/py4DSTEM/process/classification/featurization.py
@@ -477,7 +477,8 @@ def get_class_DPs(self, datacube, method, thresh):
                     datacube.data.shape[2],
                     datacube.data.shape[3],
                 )
-            except ValueError:
+            # TODO Work out if its a ValueError
+            except Exception:
                 raise ValueError(
                     "Datacube must have same R_Nx and R_Ny dimensions as Featurization instance."
                 )