Remove test repeat on test_spark_udf_spark_connect (#12303)

Signed-off-by: harupy <17039389+harupy@users.noreply.github.com>
mlflow · Jun 10, 2024 · 2b3dc0f · 2b3dc0f
1 parent 73f24fd
commit 2b3dc0f
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/pyfunc/test_spark_connect.py b/tests/pyfunc/test_spark_connect.py
@@ -24,8 +24,7 @@ def spark():
     spark.stop()
 
 
-@pytest.mark.parametrize("x", range(100))
-def test_spark_udf_spark_connect(x, spark):
+def test_spark_udf_spark_connect(spark):
     X, y = load_iris(return_X_y=True)
     model = LogisticRegression().fit(X, y)
     with mlflow.start_run():