fixes pylint error

wyli · wyli · commit 5a64744fe9c6 · 2022-04-08T08:38:09.000+01:00
Signed-off-by: Wenqi Li &lt;wenqil@nvidia.com&gt;
diff --git a/tests/ngc_mmar_loading.py b/tests/ngc_mmar_loading.py
@@ -40,7 +40,7 @@ def test_loading_mmar(self, item):
             )
             pretrained_weights = {k.split(".", 1)[1]: v for k, v in pretrained_weights["state_dict"].items()}
             sys.path.append(os.path.join(f"{item['name']}", "custom"))  # custom model folder
-            from vit_network import ViTAutoEnc
+            from vit_network import ViTAutoEnc  # pylint: disable=E0401
 
             model = ViTAutoEnc(
                 in_channels=1,

Original file line number	Diff line number	Diff line change
`@@ -40,7 +40,7 @@ def test_loading_mmar(self, item):`
`40`	`40`	`)`
`41`	`41`	`pretrained_weights = {k.split(".", 1)[1]: v for k, v in pretrained_weights["state_dict"].items()}`
`42`	`42`	`sys.path.append(os.path.join(f"{item['name']}", "custom")) # custom model folder`
`43`		`- from vit_network import ViTAutoEnc`
	`43`	`+ from vit_network import ViTAutoEnc # pylint: disable=E0401`
`44`	`44`
`45`	`45`	`model = ViTAutoEnc(`
`46`	`46`	`in_channels=1,`