diff --git a/mattport/utils/math.py b/mattport/utils/math.py index ffd66e3c0..87b571ee1 100644 --- a/mattport/utils/math.py +++ b/mattport/utils/math.py @@ -16,7 +16,7 @@ def components_from_spherical_harmonics(levels: int, directions: TensorType[..., num_components = levels**2 components = torch.zeros((*directions.shape[:-1], num_components)).to(directions.device) - assert levels > 0 and levels <= 5, f"SH levels must be in [1,4], got {levels}" + assert 1 <= levels <= 5, f"SH levels must be in [1,4], got {levels}" assert directions.shape[-1] == 3, f"Direction input should have three dimensions. Got {directions.shape[-1]}" x = directions[..., 0] diff --git a/tests/nerf/field_modules/test_encodings.py b/tests/nerf/field_modules/test_encodings.py index 1708330d8..3654daa3e 100644 --- a/tests/nerf/field_modules/test_encodings.py +++ b/tests/nerf/field_modules/test_encodings.py @@ -1,7 +1,6 @@ """ Encoding Tests """ -from multiprocessing.sharedctypes import Value import pytest import torch from mattport.nerf.field_modules import encoding