fix pyarrow

xoolive · Oct 30, 2024 · a52cd05 · a52cd05
1 parent 47e6bed
commit a52cd05
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/traffic/algorithms/openap.py b/src/traffic/algorithms/openap.py
@@ -32,7 +32,7 @@ def phases(self, twindow: int = 60) -> "Flight":
 
         fp = FlightPhase()
         fp.set_trajectory(
-            (self.data.timestamp.astype(int) // 1_000_000_000).values,
+            self.data.timestamp.dt.as_unit("s").astype(int).values,
             altitude,
             groundspeed,
             vertical_rate,

diff --git a/src/traffic/core/time.py b/src/traffic/core/time.py
@@ -1,5 +1,5 @@
 import logging
-from datetime import datetime, timezone
+from datetime import datetime
 from numbers import Real
 
 from pyopensky.time import (
@@ -53,4 +53,4 @@ def to_datetime(time: timelike) -> pd.Timestamp:
             "automatically, look at the tzinfo (resp. tz) argument of the "
             "datetime (resp. pd.Timestamp) constructor."
         )
-    return time  # type: ignore
+    return time