.

Author: sentrivana

sentry_sdk/integrations/spark/spark_driver.py

@@ -296,7 +296,7 @@ def onStageCompleted(self, stageCompleted):  # noqa: N802,N803
 
 def _get_attempt_id(stage_info):
     # type: (Any) -> Optional[int]
-    from py4j.protocol import Py4JJavaError  # type: ignore
+    from py4j.protocol import Py4JJavaError
 
     try:
         return stage_info.attemptId()
@@ -308,9 +308,4 @@ def _get_attempt_id(stage_info):
     except Py4JJavaError:
         pass
 
-    try:
-        return stage_info.currentAttemptId()
-    except Py4JJavaError:
-        pass
-
     return None