Skip to content

Commit 917d440

Browse files
author
Ainur Karimov
committed
remove redund comments
1 parent 48eddd2 commit 917d440

File tree

1 file changed

+0
-6
lines changed
  • tests/unit/datamodules/lightning_datamodules

1 file changed

+0
-6
lines changed

tests/unit/datamodules/lightning_datamodules/test_rtk.py

-6
Original file line numberDiff line numberDiff line change
@@ -206,9 +206,6 @@ def test_datamodule_description():
206206

207207
@patch("matplotlib.pyplot.show")
208208
def test_hemor_contrast(mock_show, rtk_data, tmp_path_factory):
209-
# target_dir = "./data/rtk/infer"
210-
# if os.path.exists(target_dir):
211-
# shutil.rmtree(target_dir)
212209
out_ = str(tmp_path_factory.mktemp("out"))
213210
hemorrhage_contrast(input_path=str(rtk_data), output_folder=out_)
214211
content = os.listdir(out_)
@@ -226,20 +223,17 @@ def test_hemor_contrast(mock_show, rtk_data, tmp_path_factory):
226223
def test_segm_detection_pipeline_metrics(
227224
mock_show, tmp_path_factory, task, rtk_data, rtk_downloader
228225
):
229-
230226
path = {"source": rtk_segm, "target": rtk_data}
231227
dm = DicomCocoDataModuleRTK(infer=path, transform=resize_transform)
232228
dm.setup_infer()
233229
ds = dm.predict_dataloader()
234230
ds.transform = resize_transform
235231

236232
samples_number = len(ds)
237-
238233
out_dir = tmp_path_factory.mktemp("out")
239234
for i in range(samples_number):
240235
random_numpy = np.random.randint(0, 1, [256, 256, 1])
241236
np.save(os.path.join(out_dir, f"{i}.npy"), random_numpy)
242-
243237
process_metrics(input_path=rtk_data, output_folder=out_dir)
244238
assert mock_show.call_count > 0
245239
assert rtk_downloader.call_count > 0

0 commit comments

Comments
 (0)