[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

demo.py

@@ -5,17 +5,18 @@
 def test_demo():
     import os
 
-    import b3d
-    import b3d.bayes3d as bayes3d
     import genjax
     import jax
     import jax.numpy as jnp
     import numpy as np
     import rerun as rr
-    from b3d import Mesh, Pose
     from genjax import Pytree
     from tqdm import tqdm
 
+    import b3d
+    import b3d.bayes3d as bayes3d
+    from b3d import Mesh, Pose
+
     rr.init("demo")
     rr.connect("127.0.0.1:8812")
 

demos/detector_segmenter.py

@@ -1,7 +1,6 @@
 import io
 import os
 
-import b3d
 import jax
 import jax.numpy as jnp
 import numpy
@@ -19,6 +18,8 @@
 )
 from transformers.models.detr.feature_extraction_detr import rgb_to_id
 
+import b3d
+
 processor = OwlViTProcessor.from_pretrained("google/owlvit-base-patch32")
 model = OwlViTForObjectDetection.from_pretrained("google/owlvit-base-patch32")
 

demos/differentiable_renderer/dense_fitting/standalone_nishad_demo.py

@@ -3,11 +3,12 @@
 import functools
 import os
 
-import b3d
 import jax
 import jax.numpy as jnp
 import numpy as np
 import rerun as rr
+
+import b3d
 from b3d import Pose
 
 rr.init("gradients")

demos/differentiable_renderer/gradient_based_pose_estimation.py

@@ -1,15 +1,16 @@
 import os
 from functools import partial
 
-import b3d
-import b3d.chisight.dense.differentiable_renderer as rendering
 import jax
 import jax.numpy as jnp
 import optax
 import rerun as rr
-from b3d import Mesh, Pose
 from tqdm import tqdm
 
+import b3d
+import b3d.chisight.dense.differentiable_renderer as rendering
+from b3d import Mesh, Pose
+
 rr.init("gradients")
 rr.connect("127.0.0.1:8812")
 
@@ -155,7 +156,7 @@ def render(params, mesh_params):
 
 meshes = [
     Mesh.from_obj_file(
-        os.path.join(ycb_dir, f'models/obj_{f"{id + 1}".rjust(6, "0")}.ply')
+        os.path.join(ycb_dir, f"models/obj_{f'{id + 1}'.rjust(6, '0')}.ply")
     ).scale(0.001)
     for id in all_data[0]["object_types"]
 ]

demos/differentiable_renderer/gradients_for_mug.py

@@ -1,16 +1,17 @@
 import os
 from functools import partial
 
-import b3d
-import b3d.chisight.dense.differentiable_renderer as rendering
 import jax
 import jax.numpy as jnp
 import optax
 import rerun as rr
 import trimesh
-from b3d import Pose
 from tqdm import tqdm
 
+import b3d
+import b3d.chisight.dense.differentiable_renderer as rendering
+from b3d import Pose
+
 rr.init("gradients")
 rr.connect("127.0.0.1:8812")
 

demos/differentiable_renderer/patch_tracking/demo_utils.py

@@ -1,10 +1,11 @@
 import os
 
-import b3d
-import b3d.utils as utils
 import jax
 import jax.numpy as jnp
 import trimesh
+
+import b3d
+import b3d.utils as utils
 from b3d import Pose
 
 ### Utils ###

demos/differentiable_renderer/patch_tracking/model.py

@@ -1,12 +1,12 @@
-import b3d
-import b3d.chisight.dense.differentiable_renderer as rendering
 import genjax
 import jax
 import jax.numpy as jnp
 import rerun as rr
-from b3d.modeling_utils import uniform_pose
 
+import b3d
+import b3d.chisight.dense.differentiable_renderer as rendering
 import demos.differentiable_renderer.patch_tracking.demo_utils as utils
+from b3d.modeling_utils import uniform_pose
 
 
 def normalize(v):

demos/differentiable_renderer/patch_tracking/multiple_patch_tracker_2.py

@@ -1,11 +1,11 @@
-import b3d
-import b3d.chisight.dense.patch_tracking as tracking
 import numpy as np
 import rerun as rr
-from b3d.chisight.dense.model import rr_log_uniformpose_meshes_to_image_model_trace
 from tqdm import tqdm
 
+import b3d
+import b3d.chisight.dense.patch_tracking as tracking
 import demos.differentiable_renderer.patch_tracking.demo_utils as du
+from b3d.chisight.dense.model import rr_log_uniformpose_meshes_to_image_model_trace
 
 rr.init("multiple_patch_tracking_2")
 rr.connect("127.0.0.1:8812")

demos/differentiable_renderer/patch_tracking/multiple_patch_tracking.py

@@ -2,21 +2,21 @@
 
 import os
 
-import b3d
-import b3d.chisight.dense.differentiable_renderer as r
-import b3d.chisight.dense.likelihoods as l
 import genjax
 import jax
 import jax.numpy as jnp
 import numpy as np
 import optax
 import rerun as rr
 import trimesh
-from b3d import Pose
 from tqdm import tqdm
 
+import b3d
+import b3d.chisight.dense.differentiable_renderer as r
+import b3d.chisight.dense.likelihoods as l
 import demos.differentiable_renderer.patch_tracking.demo_utils as du
 import demos.differentiable_renderer.patch_tracking.model as m
+from b3d import Pose
 
 rr.init("multiple_patch_tracking")
 rr.connect("127.0.0.1:8812")

demos/differentiable_renderer/patch_tracking/single_patch_tracking_adam.py

@@ -1,17 +1,17 @@
 ### Preliminaries ###
 
-import b3d.chisight.dense.differentiable_renderer as r
-import b3d.chisight.dense.likelihoods as l
 import genjax
 import jax
 import jax.numpy as jnp
 import optax
 import rerun as rr
-from b3d import Pose
 from tqdm import tqdm
 
+import b3d.chisight.dense.differentiable_renderer as r
+import b3d.chisight.dense.likelihoods as l
 import demos.differentiable_renderer.patch_tracking.demo_utils as du
 import demos.differentiable_renderer.patch_tracking.model as m
+from b3d import Pose
 
 rr.init("single_patch_tracking")
 rr.connect("127.0.0.1:8812")