adding KeyError to correct place ...

py4dstem · Nov 10, 2023 · dbfa1b2 · dbfa1b2
1 parent 9657c42
commit dbfa1b2
Showing 1 changed file with 2 additions and 2 deletions.
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,KeyError):
+        except ValueError:
             print("Warning: scan shape not detected. Please check/set manually.")
             R_Nx = self._guess_number_frames() // 32
             R_Ny = 1
@@ -125,7 +125,7 @@ def __init__(
             Q_Nx = gtg.allTags[".SI Image Tags.Acquisition.Parameters.Detector.height"]
             Q_Ny = gtg.allTags[".SI Image Tags.Acquisition.Parameters.Detector.width"]
         # TODO check this is the correct error type
-        except ValueError:
+        except (ValueError, KeyError):
             print("Warning: diffraction pattern shape not detected!")
             print("Assuming 1920x1792 as the diffraction pattern size!")
             Q_Nx = 1792