Skip to content

Commit 638d546

Browse files
committed
quick correct
1 parent 3b98ad9 commit 638d546

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

zenodo-tests/test_recon/test_rotation.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,19 +6,20 @@
66

77
from httomolibgpu.prep.normalize import normalize
88
from httomolibgpu.recon.rotation import find_center_vo, find_center_pc, find_center_360
9+
from conftest import force_clean_gpu_memory
910

1011

1112
# ----------------------------------------------------------#
1213
# i12_dataset1 tests
13-
def test_center_vo_i12_dataset1(i12_dataset1, ensure_clean_memory):
14+
def test_center_vo_i12_dataset1(i12_dataset1):
1415
projdata = i12_dataset1[0]
1516
flats = i12_dataset1[2]
1617
darks = i12_dataset1[3]
1718
del i12_dataset1
1819

1920
data_normalised = normalize(projdata, flats, darks, minus_log=True)
2021
del flats, darks, projdata
21-
ensure_clean_memory
22+
force_clean_gpu_memory()
2223

2324
mid_slice = data_normalised.shape[1] // 2
2425
cor = find_center_vo(

0 commit comments

Comments
 (0)