From 89b97b80d497f8045019581314dad62ae0e557c1 Mon Sep 17 00:00:00 2001 From: Loki Date: Tue, 30 Aug 2022 19:38:43 -0700 Subject: [PATCH] fix: trcomp fixtures to override default fixtures for integ tests --- tests/integ/test_training_compiler.py | 2 ++ 1 file changed, 2 insertions(+) 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