add type annotations

pyaerocom/io/readungriddedbase.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import abc
 import glob
 import logging
@@ -207,7 +209,7 @@ def read(self, vars_to_retrieve=None, files=[], first_file=None, last_file=None)
 
     ### Concrete implementations of methods that are the same for all (or most)
     # of the derived reading classes
-    def __init__(self, data_id=None, data_dir=None):
+    def __init__(self, data_id: str | None = None, data_dir: str | None = None):
         self.data = None  # object that holds the loaded data
         self._data_id = None
         self.files = []
@@ -246,7 +248,7 @@ def DATASET_PATH(self):
         return self.data_dir
 
     @property
-    def data_dir(self):
+    def data_dir(self) -> str:
         """
         str: Location of the dataset
 

pyaerocom/plugins/icos/reader.py

@@ -60,7 +60,7 @@ class ReadICOS(ReadMEP):
 
     PROVIDES_VARIABLES = list(VAR_MAPPING) + list(AUX_FUNS)
 
-    def __init__(self, data_id=None, data_dir=None):
+    def __init__(self, data_id: str | None = None, data_dir: str | None = None):
         if data_dir is None:
             data_dir = const.OBSLOCS_UNGRIDDED[const.ICOS_NAME]
 

pyaerocom/plugins/mep/reader.py

@@ -102,7 +102,7 @@ class ReadMEP(ReadUngriddedBase):
 
     PROVIDES_VARIABLES = list(VAR_MAPPING) + list(AUX_FUNS)
 
-    def __init__(self, data_id=None, data_dir=None):
+    def __init__(self, data_id: str | None = None, data_dir: str | None = None):
         if data_dir is None:
             data_dir = const.OBSLOCS_UNGRIDDED[const.MEP_NAME]
 

pyaerocom/ungriddeddata.py

@@ -471,7 +471,7 @@ def copy(self):
         return new
 
     @property
-    def contains_vars(self):
+    def contains_vars(self) -> list[str]:
         """List of all variables in this dataset"""
         return list(self.var_idx)