support ipyevents / ipycanvas import error with just a warning, fix (#…

…723)

issue with comm version

Signed-off-by: Clement Fuji Tsang <cfujitsang@nvidia.com>

ci/gitlab_jenkins_templates/ubuntu_test_CI.jenkins

@@ -86,7 +86,7 @@ spec:
         try {
           stage("Pytest") {
             sh '''
-            export KAOLIN_TEST_NVIDIFFRAST=1
+            export KAOLIN_TEST_NVDIFFRAST=1
             export KAOLIN_TEST_SHAPENETV1_PATH=/mnt/data/ci_shapenetv1
             export KAOLIN_TEST_SHAPENETV2_PATH=/mnt/data/ci_shapenetv2
             export KAOLIN_TEST_MODELNET_PATH=/mnt/data/ModelNet
@@ -100,7 +100,7 @@ spec:
         }
         try {
           stage("Dash3D") {
-            '''
+            sh '''
             pytest -s --cov=/kaolin/kaolin /kaolin/tests/integration/
             '''
           }

kaolin/visualize/ipython.py

@@ -14,17 +14,22 @@
 # limitations under the License.
 
 from abc import abstractmethod
+from io import BytesIO
 import math
+import traceback
+import warnings
 
+from PIL import Image as PILImage
 import torch
 
-from ipyevents import Event
-from ipywidgets import Output
-from ipywidgets import Image as ImageWidget
-from ipycanvas import Canvas
+try:
+    from ipyevents import Event
+    from ipywidgets import Output
+    from ipywidgets import Image as ImageWidget
+    from ipycanvas import Canvas
+except Exception as e:
+    warnings.warn('Error importing kaolin.visualize.ipython:\n' + traceback.format_exc())
 
-from io import BytesIO
-from PIL import Image as PILImage
 
 from ..render.camera import CameraExtrinsics
 

tools/viz_requirements.txt

@@ -6,3 +6,4 @@ jupyter_client<8
 pyzmq<25
 flask
 tornado
+comm>=0.1.3