star

venv/lib/python3.9/site-packages/mlflow/tracing/utils/token.py

@@ -0,0 +1,19 @@
+import contextvars
+from dataclasses import dataclass
+from typing import Optional
+
+from mlflow.entities import LiveSpan
+
+
+@dataclass
+class SpanWithToken:
+    """
+    A utility container to hold an MLflow span and its corresponding OpenTelemetry token.
+
+    The token is a special object that is generated when setting a span as active within
+    the Open Telemetry span context. This token is required when inactivate the span i.e.
+    detaching the span from the context.
+    """
+
+    span: LiveSpan
+    token: Optional[contextvars.Token] = None