Skip to content

Commit 8a1dea0

Browse files
szaherandreyvelich
authored andcommitted
Move commits from kubeflow/trainer
Signed-off-by: Andrey Velichkevich <andrey.velichkevich@gmail.com>
2 parents 410a445 + b890a2a commit 8a1dea0

File tree

371 files changed

+39471
-0
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

371 files changed

+39471
-0
lines changed

python/kubeflow/__init__.py

Whitespace-only changes.

python/kubeflow/py.typed

Whitespace-only changes.
Lines changed: 42 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,42 @@
1+
# Copyright 2024 The Kubeflow Authors.
2+
#
3+
# Licensed under the Apache License, Version 2.0 (the "License");
4+
# you may not use this file except in compliance with the License.
5+
# You may obtain a copy of the License at
6+
#
7+
# http://www.apache.org/licenses/LICENSE-2.0
8+
#
9+
# Unless required by applicable law or agreed to in writing, software
10+
# distributed under the License is distributed on an "AS IS" BASIS,
11+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+
# See the License for the specific language governing permissions and
13+
# limitations under the License.
14+
15+
16+
from __future__ import absolute_import
17+
18+
__version__ = "0.1.0"
19+
20+
# Import the Kubeflow Trainer client.
21+
from kubeflow.trainer.api.trainer_client import TrainerClient
22+
23+
# Import the Kubeflow Trainer constants.
24+
from kubeflow.trainer.constants.constants import DATASET_PATH, MODEL_PATH
25+
26+
# Import the Kubeflow Trainer types.
27+
from kubeflow.trainer.types.types import (
28+
BuiltinTrainer,
29+
CustomTrainer,
30+
DataFormat,
31+
DataType,
32+
Framework,
33+
HuggingFaceDatasetInitializer,
34+
HuggingFaceModelInitializer,
35+
Initializer,
36+
TorchTuneInstructDataset,
37+
Loss,
38+
Runtime,
39+
Trainer,
40+
TrainerType,
41+
TorchTuneConfig,
42+
)
Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
# flake8: noqa
2+
3+
# import apis into api package
4+

0 commit comments

Comments
 (0)