move setting span to _capture_exception

Author: Zenithatic

sentry_sdk/integrations/huggingface_hub.py

@@ -50,7 +50,9 @@ def setup_once() -> None:
         )
 
 
-def _capture_exception(exc: "Any") -> None:
+def _capture_exception(exc: "Any", span: "Any" = None) -> None:
+    if span is not None:
+        span.set_status(SPANSTATUS.INTERNAL_ERROR)
     event, hint = event_from_exception(
         exc,
         client_options=sentry_sdk.get_client().options,
@@ -126,8 +128,7 @@ 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.set_status(SPANSTATUS.INTERNAL_ERROR)
+                _capture_exception(e, span)
                 span.__exit__(None, None, None)
             reraise(*exc_info)