diff --git a/tests/integ/test_training_compiler.py b/tests/integ/test_training_compiler.py index 3d5fad9ea9..b42272f53b 100644 --- a/tests/integ/test_training_compiler.py +++ b/tests/integ/test_training_compiler.py @@ -26,10 +26,12 @@ from tests.integ.timeout import timeout +@pytest.fixture(scope="module") def gpu_instance_type(request): return "ml.p3.2xlarge" +@pytest.fixture(scope="module") def instance_count(request): return 1