From d71105f5e57d8bd7ae95fbd93a82cff536ba5298 Mon Sep 17 00:00:00 2001 From: Jeremy Reizenstein Date: Thu, 12 Jan 2023 14:23:39 -0800 Subject: [PATCH] lints Summary: lint fixes Reviewed By: davidsonic Differential Revision: D42451530 fbshipit-source-id: 120bdd58fc074a713895df15df4e9efa9ea0a420 --- pytorch3d/implicitron/tools/vis_utils.py | 4 ++-- tests/test_render_meshes.py | 5 ++++- 2 files changed, 6 insertions(+), 3 deletions(-) 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)