set env by default for test

AlibabaPAI · Aug 26, 2024 · 35f10b0 · 35f10b0
1 parent 8cc28b3
commit 35f10b0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/test/ds/test_bounded_dynamic.py b/test/ds/test_bounded_dynamic.py
@@ -65,7 +65,8 @@ def test_sizeEq(self):
 
 
 if __name__ == '__main__':
-  assert os.environ['USE_BOUND_FOR_SHAPE_COMPARE'] == '1'
+  os.environ['USE_BOUND_FOR_SHAPE_COMPARE'] = os.getenv(
+      'USE_BOUND_FOR_SHAPE_COMPARE', '1')
   test = unittest.main()
   # DISABLE PYTHON DISPATCHER FLAG
   del pd