use constants from HydUtils

duynguyen02 · Nov 30, 2024 · ab3fa69 · ab3fa69
1 parent 406508d
commit ab3fa69
Show file tree

Hide file tree

Showing 3 changed files with 168 additions and 125 deletions.
diff --git a/hydtank/columns_constants.py b/hydtank/columns_constants.py
diff --git a/hydtank/dataset.py b/hydtank/dataset.py
@@ -2,30 +2,38 @@
 from datetime import datetime
 from typing import List
 
+from hydutils.hyd_constants import (
+    TIMESERIES,
+    PRECIPITATION,
+    EVAPOTRANSPIRATION,
+    DISCHARGE,
+)
 import pandas as pd
 
-from hydtank.columns_constants import TIME_SERIES, PRECIPITATION, EVAPOTRANSPIRATION, DISCHARGE
-
 
 class Dataset:
-    def __init__(self, time_series: List[datetime], precipitation: List[float],
-                 evapotranspiration: List[float],
-                 discharge: List[float]):
+    def __init__(
+        self,
+        time_series: List[datetime],
+        precipitation: List[float],
+        evapotranspiration: List[float],
+        discharge: List[float],
+    ):
         self._time_series = time_series
         self._precipitation = precipitation
         self._evapotranspiration = evapotranspiration
         self._discharge = discharge
 
     def to_dataframe(self):
         dataset_dict = {
-            TIME_SERIES: self._time_series,
+            TIMESERIES: self._time_series,
             PRECIPITATION: self._precipitation,
             EVAPOTRANSPIRATION: self._evapotranspiration,
-            DISCHARGE: self._discharge
+            DISCHARGE: self._discharge,
         }
 
         df = pd.DataFrame(dataset_dict)
-        df[TIME_SERIES] = pd.to_datetime(df[TIME_SERIES])
+        df[TIMESERIES] = pd.to_datetime(df[TIMESERIES])
         return df
 
     def copy(self):