pai/job/__init__.py

#  Copyright 2024 Alibaba, Inc. or its affiliates.
#
#  Licensed under the Apache License, Version 2.0 (the "License");
#  you may not use this file except in compliance with the License.
#  You may obtain a copy of the License at
#
#       https://www.apache.org/licenses/LICENSE-2.0
#
#  Unless required by applicable law or agreed to in writing, software
#  distributed under the License is distributed on an "AS IS" BASIS,
#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#  See the License for the specific language governing permissions and
#  limitations under the License.

from ._local_training_job import LocalTrainingJob
from ._training_job import (
    AlgorithmSpec,
    Channel,
    CodeDir,
    ExperimentConfig,
    HyperParameterDefinition,
    InstanceSpec,
    ModelRecipeSpec,
    OssLocation,
    ResourceType,
    SpotSpec,
    SpotStrategy,
    TrainingJob,
    TrainingJobStatus,
    UriInput,
    UriOutput,
    UserVpcConfig,
    _TrainingJobSubmitter,
)

__all__ = [
    "TrainingJob",
    "ModelRecipeSpec",
    "TrainingJobStatus",
    "Channel",
    "HyperParameterDefinition",
    "OssLocation",
    "AlgorithmSpec",
    "CodeDir",
    "LocalTrainingJob",
    "UriOutput",
    "UserVpcConfig",
    "ExperimentConfig",
    "InstanceSpec",
    "UriInput",
    "SpotSpec",
    "ResourceType",
    "SpotStrategy",
]