bugfix

2023-09-12 15:39:02 +02:00
parent ae09fdd3d0
commit d261759dc0
1 changed files with 1 additions and 1 deletions
--- a/src/scgenerator/io.py
+++ b/src/scgenerator/io.py
@@ -233,7 +233,7 @@ class DataFile:

    def load_arrays(self, *labels: str | tuple[str, ...]) -> tuple[np.ndarray, ...]:
        raw_data = self.load_bytes()
-        extension = self.path.lower().split()[-1]
+        extension = self.path.lower().split(".")[-1]
        if extension == "npz":
            df = np.load(BytesIO(raw_data))
            return load_npz_data(df, *labels)