Report tcnn import errors (#1969)

nerfstudio/field_components/encodings.py

@@ -34,7 +34,7 @@ try:
     import tinycudann as tcnn
 
     TCNN_EXISTS = True
-except ImportError:
+except ModuleNotFoundError:
     TCNN_EXISTS = False
 
 

nerfstudio/fields/instant_ngp_field.py

@@ -38,7 +38,7 @@ from nerfstudio.fields.base_field import Field, shift_directions_for_tcnn
 
 try:
     import tinycudann as tcnn
-except ImportError:
+except ModuleNotFoundError:
     # tinycudann module doesn't exist
     pass
 

nerfstudio/fields/nerfacto_field.py

@@ -49,7 +49,7 @@ from nerfstudio.fields.base_field import Field, shift_directions_for_tcnn
 
 try:
     import tinycudann as tcnn
-except ImportError:
+except ModuleNotFoundError:
     # tinycudann module doesn't exist
     pass
 

nerfstudio/fields/nerfplayer_nerfacto_field.py

@@ -43,7 +43,7 @@ from nerfstudio.fields.base_field import Field, shift_directions_for_tcnn
 
 try:
     import tinycudann as tcnn
-except ImportError:
+except ModuleNotFoundError:
     # tinycudann module doesn't exist
     pass
 

nerfstudio/fields/nerfplayer_ngp_field.py

@@ -38,7 +38,7 @@ from nerfstudio.fields.base_field import Field, shift_directions_for_tcnn
 
 try:
     import tinycudann as tcnn
-except ImportError:
+except ModuleNotFoundError:
     # tinycudann module doesn't exist
     pass
 

nerfstudio/fields/sdf_field.py

@@ -36,7 +36,7 @@ from nerfstudio.fields.base_field import Field, FieldConfig
 
 try:
     import tinycudann as tcnn
-except ImportError:
+except ModuleNotFoundError:
     # tinycudann module doesn't exist
     pass
 

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