fix fixture

laurenyu · laurenyu · commit 9d9408f7f214 · 2020-06-19T17:55:34.000-07:00
diff --git a/tests/integ/test_knn.py b/tests/integ/test_knn.py
@@ -57,7 +57,7 @@ def test_knn_regressor(sagemaker_session, cpu_instance_type, training_set):
             assert record.label["score"] is not None
 
 
-def test_async_knn_classifier(sagemaker_session, cpu_instance_type):
+def test_async_knn_classifier(sagemaker_session, cpu_instance_type, training_set):
     job_name = unique_name_from_base("knn")
 
     with timeout(minutes=5):
