Merge pull request #127 from qitianshi/109-splitext

Fix #109
qitianshi · Apr 4, 2022 · 7a13f95 · 7a13f95
2 parents e0ae154 + acb233a
commit 7a13f95
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/analysis/oscil.py b/analysis/oscil.py
@@ -16,7 +16,7 @@ def load_npy(date: str = None):
 
     for file in os.listdir(paths.data.raw(date)):
 
-        if file.endswith(".npy"):
+        if os.path.splitext(file)[1].strip('.') == "npy":
 
             #TODO: raise FileNotFoundError