Report tcnn import errors (#1969)

nerfstudio-project · May 22, 2023 · 27e2c87 · 27e2c87
1 parent 3bc160c
commit 27e2c87
Show file tree

Hide file tree

Showing 7 changed files with 7 additions and 7 deletions.
diff --git a/nerfstudio/field_components/encodings.py b/nerfstudio/field_components/encodings.py
@@ -34,7 +34,7 @@
     import tinycudann as tcnn
 
     TCNN_EXISTS = True
-except ImportError:
+except ModuleNotFoundError:
     TCNN_EXISTS = False
 
 

diff --git a/nerfstudio/fields/instant_ngp_field.py b/nerfstudio/fields/instant_ngp_field.py
@@ -38,7 +38,7 @@
 
 try:
     import tinycudann as tcnn
-except ImportError:
+except ModuleNotFoundError:
     # tinycudann module doesn't exist
     pass
 

diff --git a/nerfstudio/fields/nerfacto_field.py b/nerfstudio/fields/nerfacto_field.py
@@ -49,7 +49,7 @@
 
 try:
     import tinycudann as tcnn
-except ImportError:
+except ModuleNotFoundError:
     # tinycudann module doesn't exist
     pass
 

diff --git a/nerfstudio/fields/nerfplayer_nerfacto_field.py b/nerfstudio/fields/nerfplayer_nerfacto_field.py
@@ -43,7 +43,7 @@
 
 try:
     import tinycudann as tcnn
-except ImportError:
+except ModuleNotFoundError:
     # tinycudann module doesn't exist
     pass
 

diff --git a/nerfstudio/fields/nerfplayer_ngp_field.py b/nerfstudio/fields/nerfplayer_ngp_field.py
@@ -38,7 +38,7 @@
 
 try:
     import tinycudann as tcnn
-except ImportError:
+except ModuleNotFoundError:
     # tinycudann module doesn't exist
     pass
 

diff --git a/nerfstudio/fields/sdf_field.py b/nerfstudio/fields/sdf_field.py
@@ -36,7 +36,7 @@
 
 try:
     import tinycudann as tcnn
-except ImportError:
+except ModuleNotFoundError:
     # tinycudann module doesn't exist
     pass
 

diff --git a/tests/field_components/test_fields.py b/tests/field_components/test_fields.py
@@ -11,7 +11,7 @@ def test_tcnn_instant_ngp_field():
     """Test the tiny-cuda-nn field"""
     try:
         import tinycudann as tcnn  # noqa: F401
-    except ImportError as e:
+    except ModuleNotFoundError as e:
         # tinycudann module doesn't exist
         print(e)
         return