fixed lint error

getsentry · Aug 27, 2024 · 1828149 · 1828149
1 parent da491d4
commit 1828149
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/sentry_sdk/integrations/spark/spark_driver.py b/sentry_sdk/integrations/spark/spark_driver.py
@@ -10,6 +10,7 @@
     from typing import Optional
 
     from sentry_sdk._types import Event, Hint
+    from pyspark import SparkContext
 
 
 class SparkIntegration(Integration):
@@ -38,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`
     """
@@ -51,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
@@ -88,7 +89,6 @@ def process_event(event, hint):
 
 def _activate_integration(sc):
     # type: (SparkContext) -> None
-    from pyspark import SparkContext
 
     _start_sentry_listener(sc)
     _set_app_properties()