fix #12 interpolate with DatetimeTZBlock

xoolive · Mar 1, 2019 · 9b85af8 · 9b85af8
1 parent 50bbc48
commit 9b85af8
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/traffic/core/flight.py b/traffic/core/flight.py
@@ -31,6 +31,11 @@
 
 # fmt: on
 
+# fix https://github.com/xoolive/traffic/issues/12
+# if pd.__version__ <= "0.24.1":
+from pandas.core.internals import Block, DatetimeTZBlock
+DatetimeTZBlock.interpolate = Block.interpolate
+
 
 def _split(data: pd.DataFrame, value, unit) -> Iterator[pd.DataFrame]:
     diff = data.timestamp.diff().values