75 lines
2.1 KiB
Python
75 lines
2.1 KiB
Python
"""
|
|
The ``mlflow.entities`` module defines entities returned by the MLflow
|
|
`REST API <../rest-api.html>`_.
|
|
"""
|
|
|
|
from mlflow.entities.assessment import (
|
|
Assessment,
|
|
AssessmentError,
|
|
AssessmentSource,
|
|
AssessmentSourceType,
|
|
)
|
|
from mlflow.entities.dataset import Dataset
|
|
from mlflow.entities.dataset_input import DatasetInput
|
|
from mlflow.entities.dataset_summary import _DatasetSummary
|
|
from mlflow.entities.document import Document
|
|
from mlflow.entities.experiment import Experiment
|
|
from mlflow.entities.experiment_tag import ExperimentTag
|
|
from mlflow.entities.file_info import FileInfo
|
|
from mlflow.entities.input_tag import InputTag
|
|
from mlflow.entities.lifecycle_stage import LifecycleStage
|
|
from mlflow.entities.metric import Metric
|
|
from mlflow.entities.model_registry import Prompt
|
|
from mlflow.entities.param import Param
|
|
from mlflow.entities.run import Run
|
|
from mlflow.entities.run_data import RunData
|
|
from mlflow.entities.run_info import RunInfo
|
|
from mlflow.entities.run_inputs import RunInputs
|
|
from mlflow.entities.run_status import RunStatus
|
|
from mlflow.entities.run_tag import RunTag
|
|
from mlflow.entities.source_type import SourceType
|
|
from mlflow.entities.span import LiveSpan, NoOpSpan, Span, SpanType
|
|
from mlflow.entities.span_event import SpanEvent
|
|
from mlflow.entities.span_status import SpanStatus, SpanStatusCode
|
|
from mlflow.entities.trace import Trace
|
|
from mlflow.entities.trace_data import TraceData
|
|
from mlflow.entities.trace_info import TraceInfo
|
|
from mlflow.entities.view_type import ViewType
|
|
|
|
__all__ = [
|
|
"Experiment",
|
|
"FileInfo",
|
|
"Metric",
|
|
"Param",
|
|
"Prompt",
|
|
"Run",
|
|
"RunData",
|
|
"RunInfo",
|
|
"RunStatus",
|
|
"RunTag",
|
|
"ExperimentTag",
|
|
"SourceType",
|
|
"ViewType",
|
|
"LifecycleStage",
|
|
"Dataset",
|
|
"InputTag",
|
|
"DatasetInput",
|
|
"RunInputs",
|
|
"Span",
|
|
"LiveSpan",
|
|
"NoOpSpan",
|
|
"SpanEvent",
|
|
"SpanStatus",
|
|
"SpanType",
|
|
"Trace",
|
|
"TraceData",
|
|
"TraceInfo",
|
|
"SpanStatusCode",
|
|
"_DatasetSummary",
|
|
"Document",
|
|
"Assessment",
|
|
"AssessmentError",
|
|
"AssessmentSource",
|
|
"AssessmentSourceType",
|
|
]
|