changing to assertion error

alex-rakowski · Nov 10, 2023 · 51f078f · 51f078f
1 parent 2212461
commit 51f078f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/py4DSTEM/datacube/virtualimage.py b/py4DSTEM/datacube/virtualimage.py
@@ -415,7 +415,7 @@ def position_detector(
                     image = self.tree(k)
                     break
                 # TODO check what error is produced when passing nonexistant key to tree
-                except (KeyError, AttributeError):
+                except AssertionError:
                     pass
             if image is None:
                 image = self[0, 0]