fix path in write_excel, write_csv

CLIMADA-project · Jan 9, 2024 · 19cae86 · 19cae86
1 parent 2f248b8
commit 19cae86
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/climada/hazard/centroids/centr.py b/climada/hazard/centroids/centr.py
@@ -721,12 +721,12 @@ def write_csv(self, file_path):
 
         Parameters
         ----------
-        file_path : str
+        file_path : str, Path
             absolute or relative file path and name to write to
         """
         LOGGER.info('Writing %s', file_path)
         df = self._centroids_to_df()
-        df.to_csv(file_path.with_suffix('.csv'), index=False)
+        df.to_csv(Path(file_path).with_suffix('.csv'), index=False)
 
 
     @classmethod
@@ -760,13 +760,13 @@ def write_excel(self, file_path):
 
         Parameters
         ----------
-        file_path : str
+        file_path : str, Path
             absolute or relative file path and name to write to
         """
         LOGGER.info('Writing %s', file_path)
         df = self._centroids_to_df()
         df.to_excel(
-            file_path.with_suffix('.xlsx'),
+            Path(file_path).with_suffix('.xlsx'),
             sheet_name=DEF_SHEET_NAME, index=False
             )