diff --git a/pytorch3d/implicitron/tools/vis_utils.py b/pytorch3d/implicitron/tools/vis_utils.py index be5eb9d9..1b3b1f15 100644 --- a/pytorch3d/implicitron/tools/vis_utils.py +++ b/pytorch3d/implicitron/tools/vis_utils.py @@ -128,8 +128,8 @@ def visualize_basics( # TODO: handle errors on the outside try: imout = {"all": torch.cat(list(imout.values()), dim=2)} - except: - print("cant cat!") + except RuntimeError as e: + print("cant cat!", e.args) for k, v in imout.items(): viz.images( diff --git a/tests/test_render_meshes.py b/tests/test_render_meshes.py index 415b3b1b..f6b91a93 100644 --- a/tests/test_render_meshes.py +++ b/tests/test_render_meshes.py @@ -1412,7 +1412,10 @@ class TestRenderMeshes(TestCaseMixin, unittest.TestCase): images = renderer(sphere_mesh) rgb = images[0, ..., :3].squeeze().cpu() - filename = f"test_simple_sphere_outside_zfar_{int(zfar)}_{rasterizer_type.__name__}.png" + filename = ( + "test_simple_sphere_outside_zfar_" + f"{int(zfar)}_{rasterizer_type.__name__}.png" + ) # Load reference image image_ref = load_rgb_image(filename, DATA_DIR)