From 68e1b5f21353d02f8a27f326409e46d129805756 Mon Sep 17 00:00:00 2001 From: hypnopump Date: Fri, 23 Jul 2021 01:09:34 +0200 Subject: [PATCH] new utils for scn --- setup.py | 2 +- tests/test_ml_utils.py | 2 -- 2 files changed, 1 insertion(+), 3 deletions(-) diff --git a/setup.py b/setup.py index 39ef396..35c583c 100644 --- 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 index 999f4a9..27b0a5b 100644 --- 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"