Fix mypy issues

tum-esm · Jun 5, 2024 · fcd6b87 · fcd6b87
1 parent 2a7ce5a
commit fcd6b87
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 5 deletions.
diff --git a/pdm.lock b/pdm.lock
diff --git a/src/export/dataframes.py b/src/export/dataframes.py
@@ -114,7 +114,7 @@ def get_sensor_dataframe(
             "utc",
             *[column_names[t] for t in all_data_types],
         ],
-        dtypes={
+        schema_overrides={
             "utc": pl.Datetime,
             **{column_names[t]: pl.Float32
                for t in all_data_types},

diff --git a/src/retrieval/utils/ils.py b/src/retrieval/utils/ils.py
@@ -34,7 +34,7 @@ def get_ils_params(serial_number: int, date: datetime.date) -> ILSParams:
             "CHANNEL2_PE",
             "VALID_SINCE",
         ],
-        dtypes={
+        schema_overrides={
             "SERIAL_NUMBER": pl.Utf8,
             "CHANNEL1_ME": pl.Float64,
             "CHANNEL1_PE": pl.Float64,

diff --git a/src/retrieval/utils/invparms_files.py b/src/retrieval/utils/invparms_files.py
@@ -46,7 +46,7 @@ def read_and_merge_invparms_files(paths: list[str]) -> Optional[pl.DataFrame]:
                         "XCH4_S5P",
                         "XCO",
                     ],
-                    dtypes={
+                    schema_overrides={
                         "JulianDate": pl.Float64,
                         "HHMMSS_ID": str,  # type: ignore
                         "SX": str,  # type: ignore

diff --git a/src/retrieval/utils/pressure_averaging.py b/src/retrieval/utils/pressure_averaging.py
@@ -48,7 +48,7 @@ def compute_mean_pressure_around_noon(
 ) -> float:
     df = pl.read_csv(
         filepath,
-        dtypes={
+        schema_overrides={
             "UTCtime___": pl.Utf8,
             "BaroYoung": pl.Float64,
         },