fix imports

docs/quickstart.rst

@@ -396,6 +396,8 @@ colour based on the vertical rate average value.
 
 .. jupyter-execute::
 
+    import pandas as pd
+
     with plt.style.context("traffic"):
         fig, ax = plt.subplots(subplot_kw=dict(projection=Lambert93()))
 

src/traffic/algorithms/navigation.py

@@ -1022,23 +1022,9 @@ def holding_pattern(
         trajectories landing at different European airports including London
         Heathrow.
 
-        .. warning::
-
-            The ``onnxruntime`` package is an optional dependency required for
-            this function to properly work. At the time being, it is not
-            available for pip with Python 3.10 but still available on
-            conda-forge servers.
-
         (new in version 2.8)
         """
-        try:
-            import onnxruntime as rt
-        except ImportError:
-            warnings.warn(
-                "Missing optional dependency: onnxruntime.\n"
-                "Retry after installing with pip or conda."
-            )
-            raise
+        import onnxruntime as rt
 
         # The following cast secures the typing
         self = cast("Flight", self)

src/traffic/core/flight.py

@@ -698,7 +698,7 @@ def label(
 
         result = self.assign(
             **dict(
-                (key, False if value is True else None)
+                (key, False if value is True else pd.NA)
                 for key, value in kwargs.items()
             )
         )

tests/test_flight.py

@@ -877,7 +877,7 @@ def test_DME_NSE_computation() -> None:
     assert_frame_equal(result_df[["NSE", "NSE_idx"]], expected, rtol=1e-3)
 
 
-@pytest.mark.skipif(version > (3, 12), reason="onnxruntime not ready for 3.12")
+# @pytest.mark.skipif(version > (3, 13), reason="onnxruntime not ready")
 def test_holding_pattern() -> None:
     holding_pattern = belevingsvlucht.holding_pattern().next()
     assert holding_pattern is not None
@@ -887,7 +887,7 @@ def test_holding_pattern() -> None:
     )
 
 
-@pytest.mark.skipif(version > (3, 12), reason="onnxruntime not ready for 3.12")
+# @pytest.mark.skipif(version > (3, 13), reason="onnxruntime not ready")
 def test_label() -> None:
     from traffic.data.datasets import landing_zurich_2019
 
@@ -909,9 +909,10 @@ def test_label() -> None:
     )
 
     assert labelled.index_max == 2
-    ils = set(ils for ils in labelled.ILS_unique if ils is not None)
+    ils = set(ils for ils in labelled.ILS_unique if not pd.isna(ils))
     assert ils == {"14", "28"}
-    assert labelled.duration_min > pd.Timedelta("2 min 30 s")
+    min_duration = labelled.data.duration.astype("timedelta64[s]").min()
+    assert min_duration > pd.Timedelta("2 min 30 s")
 
 
 def test_split_condition() -> None: