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
This commit is contained in:
Jeremy Reizenstein 2022-04-04 04:43:33 -07:00 committed by Facebook GitHub Bot
parent 2802fd9398
commit 6473aa316c

View File

@ -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