minor fixes

py4dstem · Jan 24, 2024 · cb76e24 · cb76e24
1 parent 80fabb1
commit cb76e24
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/py4DSTEM/braggvectors/braggvector_methods.py b/py4DSTEM/braggvectors/braggvector_methods.py
@@ -828,7 +828,7 @@ def plot(
         Parameters
         ----------
         index : tuple[int,int] | list[int]
-            index for which Bragg vectors to plot
+            scan position for which Bragg vectors to plot
         cal : str, optional
             Choice to plot calibrated or raw Bragg vectors must be 'raw' or 'cal', by default 'cal'
         returnfig : bool, optional
@@ -839,7 +839,8 @@ def plot(
         tuple (figure, axes)
             matplotlib figure, axes returned if `returnfig` is True
         """
-        assert cal.lower() in (
+        cal = cal.lower()
+        assert cal in (
             "cal",
             "raw",
         ), f"'cal' must be in ('cal', 'raw') {cal = } passed"