Apply suggestions from code review

Co-authored-by: Andy Ye <andyye333@gmail.com>
apache · Jun 15, 2022 · fa42b67 · fa42b67
1 parent d70eee6
commit fa42b67
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/sdks/python/pytest.ini b/sdks/python/pytest.ini
@@ -48,7 +48,7 @@ markers =
     # We run these tests with multiple major pyarrow versions (BEAM-11211)
     uses_pyarrow: tests that utilize pyarrow in some way
     uses_pytorch: tests that utilize pytorch in some way
-    uses_sklearn: collects for Sklearn RunInference integration test runs.
+    uses_sklearn: tests that utilize scikit-learn in some way
 
 # Default timeout intended for unit tests.
 # If certain tests need a different value, please see the docs on how to

diff --git a/sdks/python/test-suites/direct/common.gradle b/sdks/python/test-suites/direct/common.gradle
@@ -211,6 +211,7 @@ task torchInferenceTest {
         args '-c', ". ${envdir}/bin/activate && export FORCE_TORCH_IT=1 && ${runScriptsDir}/run_integration_test.sh $cmdArgs"
       }
     }
+
 }
 // Scikit-learn RunInference IT tests
 task sklearnInferenceTest {
-Original file line number
+Diff line change
@@ Expand Up / @@ -211,6 +211,7 @@ task torchInferenceTest { @@
             args '-c', ". ${envdir}/bin/activate && export FORCE_TORCH_IT=1 && ${runScriptsDir}/run_integration_test.sh $cmdArgs"
           }
         }
     }
     // Scikit-learn RunInference IT tests
     task sklearnInferenceTest {
@@ Expand Down @@