diff --git a/nerfstudio/field_components/encodings.py b/nerfstudio/field_components/encodings.py index 94cd1f189..71d3e4a6c 100644 --- a/nerfstudio/field_components/encodings.py +++ b/nerfstudio/field_components/encodings.py @@ -34,7 +34,7 @@ try: 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 index b75c539ed..3941e7ec9 100644 --- a/nerfstudio/fields/instant_ngp_field.py +++ b/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 diff --git a/nerfstudio/fields/nerfacto_field.py b/nerfstudio/fields/nerfacto_field.py index b1b1c2cfb..e828ac664 100644 --- a/nerfstudio/fields/nerfacto_field.py +++ b/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 diff --git a/nerfstudio/fields/nerfplayer_nerfacto_field.py b/nerfstudio/fields/nerfplayer_nerfacto_field.py index 0dc4ba4ce..759806f84 100644 --- a/nerfstudio/fields/nerfplayer_nerfacto_field.py +++ b/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 diff --git a/nerfstudio/fields/nerfplayer_ngp_field.py b/nerfstudio/fields/nerfplayer_ngp_field.py index 803637729..38f70e547 100644 --- a/nerfstudio/fields/nerfplayer_ngp_field.py +++ b/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 diff --git a/nerfstudio/fields/sdf_field.py b/nerfstudio/fields/sdf_field.py index 031614f2f..a6d222933 100644 --- a/nerfstudio/fields/sdf_field.py +++ b/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 diff --git a/tests/field_components/test_fields.py b/tests/field_components/test_fields.py index 4714ed4d9..9e5611fcc 100644 --- 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