File tree 1 file changed +4
-3
lines changed
1 file changed +4
-3
lines changed Original file line number Diff line number Diff line change 10
10
)
11
11
from sentry_sdk .integrations .spark .spark_worker import SparkWorkerIntegration
12
12
13
- from pyspark import SparkContext
13
+ from pyspark import SparkConf , SparkContext
14
14
15
15
from py4j .protocol import Py4JJavaError
16
16
@@ -25,12 +25,13 @@ def sentry_init_with_reset(sentry_init):
25
25
from sentry_sdk .integrations import _processed_integrations
26
26
27
27
yield lambda : sentry_init (integrations = [SparkIntegration ()])
28
- _processed_integrations .remove ("spark" )
28
+ _processed_integrations .discard ("spark" )
29
29
30
30
31
31
@pytest .fixture (scope = "function" )
32
32
def 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" )
34
35
SparkContext ._active_spark_context .stop ()
35
36
36
37
You can’t perform that action at this time.
0 commit comments