From 6473aa316c18786b043292d3e0f7d5ecb640615b Mon Sep 17 00:00:00 2001 From: Jeremy Reizenstein Date: Mon, 4 Apr 2022 04:43:33 -0700 Subject: [PATCH] avoid visdom import in tests Summary: This might make the implicitron tests work better on RE. Reviewed By: davnov134 Differential Revision: D35283131 fbshipit-source-id: 4dda9684f632ab6e9cebcbf1e6e4a8243ec00c85 --- tests/implicitron/test_dataset_visualize.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tests/implicitron/test_dataset_visualize.py b/tests/implicitron/test_dataset_visualize.py index e1be7aeb..a3b3e7d3 100644 --- a/tests/implicitron/test_dataset_visualize.py +++ b/tests/implicitron/test_dataset_visualize.py @@ -16,7 +16,10 @@ from pytorch3d.implicitron.dataset.implicitron_dataset import ImplicitronDataset from pytorch3d.implicitron.dataset.visualize import get_implicitron_sequence_pointcloud from pytorch3d.implicitron.tools.point_cloud_utils import render_point_cloud_pytorch3d from pytorch3d.vis.plotly_vis import plot_scene -from visdom import Visdom + + +if os.environ.get("INSIDE_RE_WORKER") is None: + from visdom import Visdom if os.environ.get("FB_TEST", False): from .common_resources import get_skateboard_data