File tree Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Original file line number Diff line number Diff line change 1010)
1111from sentry_sdk .integrations .spark .spark_worker import SparkWorkerIntegration
1212
13- from pyspark import SparkContext
13+ from pyspark import SparkConf , SparkContext
1414
1515from py4j .protocol import Py4JJavaError
1616
@@ -25,12 +25,13 @@ def sentry_init_with_reset(sentry_init):
2525 from sentry_sdk .integrations import _processed_integrations
2626
2727 yield lambda : sentry_init (integrations = [SparkIntegration ()])
28- _processed_integrations .remove ("spark" )
28+ _processed_integrations .discard ("spark" )
2929
3030
3131@pytest .fixture (scope = "function" )
3232def create_spark_context ():
33- yield lambda : SparkContext (appName = "Testing123" )
33+ conf = SparkConf ().set ("spark.driver.bindAddress" , "127.0.0.1" )
34+ yield lambda : SparkContext (conf = conf , appName = "Testing123" )
3435 SparkContext ._active_spark_context .stop ()
3536
3637
You can’t perform that action at this time.
0 commit comments