Skip to content

Commit

Permalink
update based on comments
Browse files Browse the repository at this point in the history
Signed-off-by: Wenqi Li <wenqil@nvidia.com>
  • Loading branch information
wyli committed Oct 4, 2021
1 parent 82c77a1 commit 4889f65
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion tests/ngc_mmar_loading.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,7 @@
import os
import unittest

import torch
from parameterized import parameterized

from monai.apps.mmars import MODEL_DESC, load_from_mmar
Expand All @@ -26,7 +27,7 @@ def setUp(self):
@parameterized.expand((item,) for item in MODEL_DESC)
def test_loading_mmar(self, item):
pretrained_model = load_from_mmar(item=item, mmar_dir="./", map_location="cpu")
print(pretrained_model)
self.assertTrue(isinstance(pretrained_model, torch.nn.Modules))

def tearDown(self):
print(os.listdir(self.test_dir))
Expand Down

0 comments on commit 4889f65

Please sign in to comment.