new utils for scn

EleutherAI · Jul 22, 2021 · 68e1b5f · 68e1b5f
1 parent 1d2bda0
commit 68e1b5f
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 setup(
   name = 'mp-nerf',
   packages = find_packages(),
-  version = '0.1.12',
+  version = '1.0.3',
   license='MIT',
   description = 'MP-NeRF: Massively Parallel Natural Extension of Reference Frame',
   author = 'Eric Alcaide',

diff --git a/tests/test_ml_utils.py b/tests/test_ml_utils.py
@@ -29,8 +29,6 @@ def test_rename_symmetric_atoms():
     cloud_mask = scn_cloud_mask(seq_list[0]).unsqueeze(0)
     pred_feats = torch.randn(1, 16, 14, 16)
 
-    print(cloud_mask.shape)
-
     renamed = rename_symmetric_atoms(pred_coors, true_coors, seq_list, cloud_mask, pred_feats=pred_feats)
     assert renamed[0].shape == pred_coors.shape and renamed[1].shape == pred_feats.shape, "Shapes don't match"