Fix flaky tests

zou3519 · zou3519 · commit ae4b3bdb3aae · 2021-12-13T17:18:12.000-08:00
diff --git a/test/test_pythonkey.py b/test/test_pythonkey.py
@@ -220,6 +220,7 @@ def f(x):
     xfail('randint_like'), # randomness
     skip('new_empty'), # nondeterministic
     skip('empty_like'), # nondeterministic
+    skip('linalg.lstsq', 'grad_oriented'), # flaky
 }
 class TestPythonKeyOperatorsOpInfo(TestCase):
     @ops(functorch_lagging_op_db + additional_op_db, allowed_dtypes=(torch.float,))

Original file line number	Diff line number	Diff line change
`@@ -220,6 +220,7 @@ def f(x):`
`220`	`220`	`xfail('randint_like'), # randomness`
`221`	`221`	`skip('new_empty'), # nondeterministic`
`222`	`222`	`skip('empty_like'), # nondeterministic`
	`223`	`+ skip('linalg.lstsq', 'grad_oriented'), # flaky`
`223`	`224`	`}`
`224`	`225`	`class TestPythonKeyOperatorsOpInfo(TestCase):`
`225`	`226`	`@ops(functorch_lagging_op_db + additional_op_db, allowed_dtypes=(torch.float,))`