fix: solve #noqa's

SainsburyWellcomeCentre · Nov 13, 2024 · 6279bda · 6279bda
1 parent 6be26d1
commit 6279bda
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/aeon/__init__.py b/aeon/__init__.py
@@ -12,4 +12,6 @@
     del version, PackageNotFoundError
 
 # Set functions available directly under the 'aeon' top-level namespace
-from aeon.io.api import load as load  # noqa: PLC0414
+from aeon.io.api import load
+
+__all__ = ["load"]
diff --git a/aeon/dj_pipeline/tracking.py b/aeon/dj_pipeline/tracking.py
@@ -370,7 +370,8 @@ def compute_distance(position_df, target, xcol="x", ycol="y"):
         xcol (str): x column name in ``position_df``. Default is 'x'.
         ycol (str): y column name in ``position_df``. Default is 'y'.
     """
-    if len(target) != 2:  # noqa PLR2004
+    COORDS = 2 # x, y
+    if len(target) != COORDS:
         raise ValueError("Target must be a list of tuple of length 2.")
     return np.sqrt(np.square(position_df[[xcol, ycol]] - target).sum(axis=1))