Skip to content

Commit c498a3b

Browse files
committed
fix format
Signed-off-by: YunLiu <55491388+KumoLiu@users.noreply.github.com>
1 parent ba8fc01 commit c498a3b

File tree

1 file changed

+1
-8
lines changed

1 file changed

+1
-8
lines changed

tests/bundle/test_bundle_download.py

Lines changed: 1 addition & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -333,7 +333,6 @@ def test_load_weights(self, bundle_files, bundle_name, repo, device, model_file)
333333
output_3 = model_3.forward(input_tensor)
334334
assert_allclose(output_3, expected_output, atol=1e-4, rtol=1e-4, type_test=False)
335335

336-
337336
@parameterized.expand([TEST_CASE_8])
338337
@skip_if_quick
339338
@skipUnless(has_huggingface_hub, "Requires `huggingface_hub`.")
@@ -342,13 +341,7 @@ def test_load_weights_with_net_override(self, bundle_name, device, net_override)
342341
# download bundle, and load weights from the downloaded path
343342
with tempfile.TemporaryDirectory() as tempdir:
344343
# load weights
345-
model = load(
346-
name=bundle_name,
347-
bundle_dir=tempdir,
348-
source="monaihosting",
349-
progress=False,
350-
device=device,
351-
)
344+
model = load(name=bundle_name, bundle_dir=tempdir, source="monaihosting", progress=False, device=device)
352345

353346
# prepare data and test
354347
input_tensor = torch.rand(1, 1, 96, 96, 96).to(device)

0 commit comments

Comments
 (0)