ruff checked

Author: mirkoklukas

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 (

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
 
 

tests/sama4d/video_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
 
 

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 ###

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
 
 

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