diff --git a/python/pyspark/sql/tests/connect/client/test_artifact.py b/python/pyspark/sql/tests/connect/client/test_artifact.py index 4aec179bcf2a..d45230e926b1 100644 --- a/python/pyspark/sql/tests/connect/client/test_artifact.py +++ b/python/pyspark/sql/tests/connect/client/test_artifact.py @@ -390,6 +390,12 @@ def test_cache_artifact(self): class LocalClusterArtifactTests(ReusedConnectTestCase, ArtifactTestsMixin): + @classmethod + def conf(cls): + return ( + super().conf().set("spark.driver.memory", "512M").set("spark.executor.memory", "512M") + ) + @classmethod def root(cls): # In local cluster, we can mimic the production usage. @@ -397,7 +403,7 @@ def root(cls): @classmethod def master(cls): - return "local-cluster[2,2,1024]" + return "local-cluster[2,2,512]" if __name__ == "__main__":