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