add py3only fixture for sm perf test

aws · Sep 11, 2020 · 867d394 · 867d394
1 parent 5dd2de9
commit 867d394
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/dlc_tests/benchmark/sagemaker/mxnet/training/test_performance_mxnet_sm_training.py b/test/dlc_tests/benchmark/sagemaker/mxnet/training/test_performance_mxnet_sm_training.py
@@ -13,7 +13,7 @@
 # This test can also be performed for 1 node, but it takes a very long time, and CodeBuild job may expire before the
 # test ends.
 @pytest.mark.parametrize("num_nodes", [4], indirect=True)
-def test_mxnet_sagemaker_training_performance(mxnet_training, num_nodes, region, gpu_only):
+def test_mxnet_sagemaker_training_performance(mxnet_training, num_nodes, region, gpu_only, py3_only):
     """
     Run MX sagemaker training performance test