dedebenui/scgenerator
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

bugfix

Browse Source
This commit is contained in:
Benoît Sierro
2023-09-12 15:39:02 +02:00
parent ae09fdd3d0
commit d261759dc0
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/scgenerator/io.py
View File

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