Update main.py

gleasonw · Jan 23, 2024 · a094724 · a094724
1 parent d791dd3
commit a094724
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/main.py b/main.py
@@ -735,10 +735,10 @@ def calc_ratio(row):
 
     def get_unix_timestamp(row) -> int:
         year = int(row.get("annee", 0))
-        month = int(row.get("mois", 0)) - 1 if row.get("mois") else 0
+        month = int(row.get("mois", 1))
 
-        dt = datetime(year, month + 1, 1) if month >= 0 else datetime(year, 1, 1)
-        return int((dt - datetime(1970, 1, 1)).total_seconds())
+        dt = datetime(year, month, 1)
+        return dt.timestamp()
 
     data = series_dataframe.apply(
         lambda row: (get_unix_timestamp(row), row["ratio"]), axis=1