address ruff complaints

Author: littleredcomputer

bayes3d/__init__.py

@@ -4,7 +4,7 @@
 
 from importlib import metadata
 
-from . import colmap, distributions, scene_graph, utils
+from . import Renderer, colmap, distributions, scene_graph, utils
 from .camera import *
 from .likelihood import *
 from .renderer import *

bayes3d/genjax/model.py

@@ -302,4 +302,4 @@ def update_address(trace, key, address, value):
         key,
         genjax.choice_map({address: value}),
         tuple(map(lambda v: Diff(v, UnknownChange), trace.args)),
-    )[0]
+    )[0]

bayes3d/viz/viz.py

@@ -274,7 +274,9 @@ def hvstack_images(images, h, w, border=10, bg_color=(255, 255, 255)):
     images_to_vstack = []
 
     for row_idx in range(h):
-        hstacked_row = hstack_images(images[row_idx * w : (row_idx + 1) * w], border=border, bg_color=bg_color)
+        hstacked_row = hstack_images(
+            images[row_idx * w : (row_idx + 1) * w], border=border, bg_color=bg_color
+        )
         images_to_vstack.append(hstacked_row)
 
     return vstack_images(images_to_vstack, border=border, bg_color=bg_color)

demo.py

@@ -7,7 +7,6 @@
 from scipy.spatial.transform import Rotation as R
 
 import bayes3d as b
-from bayes3d import distributions
 
 # Can be helpful for debugging:
 # jax.config.update('jax_enable_checks', True)