move span status setting to exit()

Author: Zenithatic

sentry_sdk/integrations/huggingface_hub.py

@@ -6,7 +6,7 @@
 import sentry_sdk
 from sentry_sdk.ai.monitoring import record_token_usage
 from sentry_sdk.ai.utils import set_data_normalized
-from sentry_sdk.consts import OP, SPANDATA, SPANSTATUS
+from sentry_sdk.consts import OP, SPANDATA
 from sentry_sdk.integrations import DidNotEnable, Integration
 from sentry_sdk.scope import should_send_default_pii
 from sentry_sdk.utils import (
@@ -50,9 +50,7 @@ def setup_once() -> None:
         )
 
 
-def _capture_exception(exc: "Any", span: "Any" = None) -> None:
-    if span is not None:
-        span.set_status(SPANSTATUS.INTERNAL_ERROR)
+def _capture_exception(exc: "Any") -> None:
     event, hint = event_from_exception(
         exc,
         client_options=sentry_sdk.get_client().options,
@@ -128,8 +126,8 @@ def new_huggingface_task(*args: "Any", **kwargs: "Any") -> "Any":
         except Exception as e:
             exc_info = sys.exc_info()
             with capture_internal_exceptions():
-                _capture_exception(e, span)
-                span.__exit__(None, None, None)
+                _capture_exception(e)
+                span.__exit__(*exc_info)
             reraise(*exc_info)
 
         # Output attributes