chore: import upstream snapshot with attribution

python/ray/data/_internal/usage/__init__.py

@@ -0,0 +1,43 @@
+import logging
+from typing import TYPE_CHECKING
+
+from ray.data._internal.usage.collector import (
+    EnvInfo,
+    Issue,
+    LogicalOp,
+    OpConfig,
+    PipelinePerf,
+    PlanNode,
+    UsageInfo,
+    WorkloadInfo,
+    record_usage_info,
+)
+from ray.data._internal.usage.execution_callback import UsageCallback
+
+if TYPE_CHECKING:
+    from ray.data._internal.logical.interfaces.logical_plan import LogicalPlan
+
+logger = logging.getLogger(__name__)
+
+
+def create_usage_callback(logical_plan: "LogicalPlan") -> UsageCallback:
+    """Create the usage callback for an execution.
+
+    Factory method to return a ``UsageCallback`` object.
+    """
+    return UsageCallback(logical_plan)
+
+
+__all__ = [
+    "EnvInfo",
+    "Issue",
+    "LogicalOp",
+    "OpConfig",
+    "PipelinePerf",
+    "PlanNode",
+    "UsageCallback",
+    "UsageInfo",
+    "WorkloadInfo",
+    "create_usage_callback",
+    "record_usage_info",
+]