From 58dd2a7eee4a3389b3b52ee1b8c3bb787d9e8293 Mon Sep 17 00:00:00 2001 From: Matt Tancik Date: Mon, 16 May 2022 13:36:40 -0700 Subject: [PATCH] lint --- mattport/utils/math.py | 2 +- tests/nerf/field_modules/test_encodings.py | 1 - 2 files changed, 1 insertion(+), 2 deletions(-) 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