Fix data loading

iiasa · Nov 25, 2024 · 1db5a56 · 1db5a56
1 parent 52614b3
commit 1db5a56
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pysquirrel/core.py b/pysquirrel/core.py
@@ -138,7 +138,7 @@ def _load(self) -> None:
 
         for data_file in os.listdir(DATA_PATH):
             for region_type, cls in region_class.items():
-                if data_file.startswith(region_type):
+                if data_file.startswith(region_type) and data_file.endswith("yaml"):
                     with open(DATA_PATH / data_file, "r", encoding="utf8") as f:
                         data = yaml.safe_load(f)
                     for region in data: