From 5e28d2fc1e00f659c36862fef30b75abf9786da8 Mon Sep 17 00:00:00 2001 From: Wenqi Li Date: Thu, 7 Oct 2021 10:30:09 +0100 Subject: [PATCH] fixes mmar tests Signed-off-by: Wenqi Li --- tests/ngc_mmar_loading.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/ngc_mmar_loading.py b/tests/ngc_mmar_loading.py index 8aa3859d87..c1ed22de5d 100644 --- a/tests/ngc_mmar_loading.py +++ b/tests/ngc_mmar_loading.py @@ -27,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") - self.assertTrue(isinstance(pretrained_model, torch.nn.Modules)) + self.assertTrue(isinstance(pretrained_model, torch.nn.Module)) def tearDown(self): print(os.listdir(self.test_dir))