adding KeyError

py4dstem · Nov 10, 2023 · 9657c42 · 9657c42
1 parent dd1a74b
commit 9657c42
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/py4DSTEM/io/filereaders/read_K2.py b/py4DSTEM/io/filereaders/read_K2.py
@@ -115,7 +115,7 @@ def __init__(
         try:
             R_Ny = gtg.allTags[".SI Dimensions.Size Y"]
             R_Nx = gtg.allTags[".SI Dimensions.Size X"]
-        except ValueError:
+        except (ValueError,KeyError):
             print("Warning: scan shape not detected. Please check/set manually.")
             R_Nx = self._guess_number_frames() // 32
             R_Ny = 1