test fix 2

voznesenskym · voznesenskym · commit bc5c9bf1176c · 2023-02-02T08:21:06.000Z
diff --git a/test/dynamo/test_export.py b/test/dynamo/test_export.py
@@ -98,7 +98,7 @@ def func(x):
         for guard in out_guards:
             if guard.source == GuardSource.SHAPE_ENV:
                 hit = True
-                self.assertTrue("x.size()[0] <= 10" in guard.code_list[0])
+                self.assertTrue("x.size()[0] <= 10" in guard.code_list)
 
         self.assertTrue(hit)
 
