Better handling of file paths in diffread

LaurentRDC · Nov 12, 2020 · 9761c46 · 9761c46
1 parent 9c5dd19
commit 9761c46
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/skued/io/io.py b/skued/io/io.py
@@ -36,11 +36,11 @@ def diffread(fname):
     skimage.io.imread : load images from files
     skued.mibread : Read single- and multi-image Merlin Image Binary files
     """
-    fname = str(fname)  # In case of pathlib.Path
+    fname = Path(fname)
 
-    if fname.endswith(".mib"):
+    if fname.suffix == ".mib":
         return mibread(fname)
-    elif fname.endswith((".dm3", ".dm4")):
+    elif fname.suffix in {".dm3", ".dm4"}:
         return dmread(fname)
 
     # Worst-case scenario, we need skimage.io