ruff checked

probcomp · Aug 22, 2024 · 3286365 · 3286365
1 parent 635b9de
commit 3286365
Show file tree

Hide file tree

Showing 6 changed files with 8 additions and 6 deletions.
diff --git a/tests/sama4d/video_to_tracks/from_initialization/registration.py b/tests/sama4d/video_to_tracks/from_initialization/registration.py
@@ -2,9 +2,10 @@
 This file registers a default set of tasks and solvers for the video to keypoint tracks task class.
 """
 
-import b3d
 import jax.numpy as jnp
 
+import b3d
+
 from ...data_curation import get_loaders_for_all_curated_scenes
 from .keypoint_tracking_task import KeypointTrackingTask
 from .solvers.particle_system_patch_tracking_solver import (

diff --git a/tests/sama4d/video_to_tracks/from_initialization/solvers/twod/single_patch_tracker.py b/tests/sama4d/video_to_tracks/from_initialization/solvers/twod/single_patch_tracker.py
@@ -1,11 +1,11 @@
-import b3d
 import jax
 import jax.numpy as jnp
 import matplotlib.patches as mpatches
 import matplotlib.pyplot as plt
 from matplotlib.animation import FFMpegWriter, FuncAnimation
 from matplotlib.gridspec import GridSpec
 
+import b3d
 from tests.common.solver import Solver
 
 

diff --git a/..._to_tracks/from_initialization/solvers/twod/single_patch_tracker_with_reinitialization.py b/..._to_tracks/from_initialization/solvers/twod/single_patch_tracker_with_reinitialization.py
@@ -1,11 +1,11 @@
-import b3d
 import jax
 import jax.numpy as jnp
 import matplotlib.patches as mpatches
 import matplotlib.pyplot as plt
 from matplotlib.animation import FFMpegWriter, FuncAnimation
 from matplotlib.gridspec import GridSpec
 
+import b3d
 from tests.common.solver import Solver
 
 

diff --git a/tests/sama4d/video_to_tracks/solvers/conv_with_reinstantiation.py b/tests/sama4d/video_to_tracks/solvers/conv_with_reinstantiation.py
@@ -1,6 +1,6 @@
 import jax
-from b3d.chisight.patch_tracking_2d.patch_tracker import PatchTracker2D
 
+from b3d.chisight.patch_tracking_2d.patch_tracker import PatchTracker2D
 from tests.common.solver import Solver
 
 ### Solver for VideoToTracksTask ###

diff --git a/tests/sama4d/video_to_tracks/video_to_tracks_task.py b/tests/sama4d/video_to_tracks/video_to_tracks_task.py
@@ -1,13 +1,13 @@
 import warnings
 from typing import Callable
 
-import b3d
 import jax.numpy as jnp
 import matplotlib.animation as animation
 import matplotlib.pyplot as plt
 import numpy as np
 import rerun as rr
 
+import b3d
 from tests.common.task import Task
 
 

diff --git a/tests/test_2d_patch_tracker.py b/tests/test_2d_patch_tracker.py
@@ -1,5 +1,6 @@
-import b3d
 import jax
+
+import b3d
 from b3d.chisight.patch_tracking_2d.patch_tracker import PatchTracker2D