Update data.py

sparks-baird · Jun 23, 2022 · adad563 · adad563
1 parent db1013a
commit adad563
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/xtal2png/utils/data.py b/src/xtal2png/utils/data.py
@@ -219,7 +219,7 @@ def get_image_mode(d: np.ndarray) -> str:
     --------
     >>> d = np.zeros((1, 64, 64), dtype=np.uint8) # grayscale image
     >>> mode = get_image_mode(d)
-    OUTPUT
+    "L"
     """
     if d.ndim != 3:
         raise ValueError("expected an array with 3 dimensions, received {d.ndim} dims")