Small changes

metno · Jul 26, 2024 · 077a449 · 077a449
1 parent 6032f6d
commit 077a449
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/pyaro_readers/eeareader/EEATimeseriesReader.py b/src/pyaro_readers/eeareader/EEATimeseriesReader.py
@@ -52,8 +52,12 @@ def __init__(
             )
         files = self._create_file_list(filename, species)
         lf = polars.scan_parquet(files)
+        df = lf.collect()
+        df = df.with_columns(
+            (polars.col("Samplingpoint").str.extract(r"(.*)/.*")).alias("Country Code")
+        )
 
-        return lf
+        return df
         # folder_list = self._create_file_list(filename, species_ids)
 
     def _create_file_list(self, root: Path, species: list[str]):
@@ -106,7 +110,7 @@ def is_valid_url(self, url):
 
 
 if __name__ == "__main__":
-    filters = {"variables": {"include": ["SO2", "PM10"]}}
+    filters = {"variables": {"include": ["SO2", "PM10", "PM2.5"]}}
     reader = EEATimeseriesReader(
         "/home/danielh/Documents/pyaerocom/pyaro-readers/src/pyaro_readers/eeareader/data",
         filters=filters,