Mark test_one_shot_ks_perf_sensitivity as flaky (#367)

dbarbuzzi · markurtz · bfineran · web-flow · commit f39f04be90da · 2021-08-25T14:32:26.000-04:00
Co-authored-by: Mark Kurtz &lt;mark@neuralmagic.com&gt;
Co-authored-by: Benjamin Fineran &lt;bfineran@users.noreply.github.com&gt;
diff --git a/tests/sparseml/onnx/optim/test_sensitivity_ks.py b/tests/sparseml/onnx/optim/test_sensitivity_ks.py
@@ -221,6 +221,7 @@ def test_one_shot_ks_loss_sensitivity(
 @pytest.mark.skipif(
     deepsparse is None, reason="deepsparse is not installed on the system"
 )
+@flaky(max_runs=2, min_passes=1)
 def test_one_shot_ks_perf_sensitivity(
     onnx_models_with_analysis: OnnxModelAnalysisFixture,
 ):

Original file line number	Diff line number	Diff line change
`@@ -221,6 +221,7 @@ def test_one_shot_ks_loss_sensitivity(`
`221`	`221`	`@pytest.mark.skipif(`
`222`	`222`	`deepsparse is None, reason="deepsparse is not installed on the system"`
`223`	`223`	`)`
	`224`	`+@flaky(max_runs=2, min_passes=1)`
`224`	`225`	`def test_one_shot_ks_perf_sensitivity(`
`225`	`226`	`onnx_models_with_analysis: OnnxModelAnalysisFixture,`
`226`	`227`	`):`