423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-08-02 11:52:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

avoid visdom import in tests

Browse Source 
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
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tests/implicitron/test_dataset_visualize.py
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.dataset.visualize import get_implicitron_sequence_pointcloud
from pytorch3d.implicitron.tools.point_cloud_utils import render_point_cloud_pytorch3d from pytorch3d.implicitron.tools.point_cloud_utils import render_point_cloud_pytorch3d
from pytorch3d.vis.plotly_vis import plot_scene 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): if os.environ.get("FB_TEST", False):
from .common_resources import get_skateboard_data from .common_resources import get_skateboard_data