Skip to content

Commit 40c611f

Browse files
author
Anurag Dixit
committed
fix: Disabled mobilenet_v2 test for DLFW CI
Signed-off-by: Anurag Dixit <[email protected]>
1 parent a10613e commit 40c611f

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/py/test_api.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -508,7 +508,8 @@ def test_suite():
508508
suite = unittest.TestSuite()
509509
suite.addTest(unittest.makeSuite(TestLoggingAPIs))
510510
suite.addTest(TestCompile.parametrize(TestCompile, model=models.resnet18(pretrained=True)))
511-
suite.addTest(TestCompile.parametrize(TestCompile, model=models.mobilenet_v2(pretrained=True)))
511+
# Disabling mobilenet_v2 test due to https://nvbugs/3433655
512+
# suite.addTest(TestCompile.parametrize(TestCompile, model=models.mobilenet_v2(pretrained=True)))
512513
suite.addTest(TestCompileHalf.parametrize(TestCompileHalf, model=models.resnet18(pretrained=True)))
513514
suite.addTest(TestCompileHalfDefault.parametrize(TestCompileHalfDefault, model=models.resnet18(pretrained=True)))
514515
suite.addTest(TestPTtoTRTtoPT.parametrize(TestPTtoTRTtoPT, model=models.resnet18(pretrained=True)))

0 commit comments

Comments
 (0)