diff --git a/sentry_sdk/integrations/spark/spark_driver.py b/sentry_sdk/integrations/spark/spark_driver.py index 905ac41131..13d568009a 100644 --- a/sentry_sdk/integrations/spark/spark_driver.py +++ b/sentry_sdk/integrations/spark/spark_driver.py @@ -39,7 +39,7 @@ def _set_app_properties(): def _start_sentry_listener(sc): - # type: (Any) -> None + # type: (SparkContext) -> None """ Start java gateway server to add custom `SparkListener` """ @@ -52,7 +52,7 @@ def _start_sentry_listener(sc): def _add_event_processor(sc): - # type: (Any) -> None + # type: (SparkContext) -> None scope = sentry_sdk.get_isolation_scope() @scope.add_event_processor