From ba1d95990a6de14d3bfc21ab65f332c092114ec5 Mon Sep 17 00:00:00 2001 From: fangyh20 Date: Mon, 27 Oct 2025 11:24:16 -0700 Subject: [PATCH] test: enable add artifacts test --- tests/integration/test_session.py | 8 +------- 1 file changed, 1 insertion(+), 7 deletions(-) diff --git a/tests/integration/test_session.py b/tests/integration/test_session.py index d39292a..f96467d 100644 --- a/tests/integration/test_session.py +++ b/tests/integration/test_session.py @@ -372,14 +372,8 @@ def test_create_spark_session_with_session_template_and_user_provided_dataproc_c assert DataprocSparkSession._active_s8s_session_uuid is None -@pytest.mark.skip( - reason="Skipping PyPI package installation test since it's not supported yet" -) def test_add_artifacts_pypi_package(): - """Test adding PyPI packages as artifacts to a Spark session. - - Note: Skipped in CI due to infrastructure issues with PyPI package installation. - """ + """Test adding PyPI packages as artifacts to a Spark session.""" connect_session = DataprocSparkSession.builder.getOrCreate() from pyspark.sql.connect.functions import udf, sum from pyspark.sql.types import IntegerType