element wise volume alignment

flatironinstitute · Aug 20, 2024 · 94b07dd · 94b07dd
1 parent 5572d97
commit 94b07dd
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/cryo_challenge/_preprocessing/align_utils.py b/src/cryo_challenge/_preprocessing/align_utils.py
@@ -135,11 +135,10 @@ def align_submission(
     volumes (torch.Tensor): aligned submission volumes
     """
     for i in range(len(volumes)):
-        print('aligning ' + str(i) + 'th volume' )
-        obj_vol = volumes[i].numpy().astype(np.float32)
+        obj_vol = volumes[i].numpy().astype(np.float32).copy()
 
         obj_vol = Volume(obj_vol / obj_vol.sum())
-        ref_vol = Volume(ref_volume / ref_volume.sum())
+        ref_vol = Volume(ref_volume.copy() / ref_volume.sum())
 
         _, R_est = align_BO(
             ref_vol,