From cbc85ef450721a5862357ec0a4954f3addee07e2 Mon Sep 17 00:00:00 2001 From: Matt Tancik Date: Fri, 29 Apr 2022 15:44:19 -0700 Subject: [PATCH] Fix test --- tests/nerf/test_sampler.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/nerf/test_sampler.py b/tests/nerf/test_sampler.py index 8e884a4be..167bda16b 100644 --- a/tests/nerf/test_sampler.py +++ b/tests/nerf/test_sampler.py @@ -21,7 +21,7 @@ def test_uniform_sampler(): ray_samples = sampler(ray_bundle) - assert ray_samples.bins.shape[-1] == num_samples + 1 + assert ray_samples.positions.shape[-2] == num_samples # TODO Tancik: Add more precise tests