diff --git a/test/test_ops.py b/test/test_ops.py index 26d13bbe208..dae2745a166 100644 --- a/test/test_ops.py +++ b/test/test_ops.py @@ -2035,4 +2035,5 @@ def test_is_leaf_node(self, dim, p, block_size, inplace): if __name__ == "__main__": - pytest.main([__file__]) + import sys + pytest.main([__file__] + sys.argv)