mlops-katib 0.19.0__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- mlops_katib-0.19.0/PKG-INFO +49 -0
- mlops_katib-0.19.0/README.md +125 -0
- mlops_katib-0.19.0/kubeflow/__init__.py +1 -0
- mlops_katib-0.19.0/kubeflow/katib/__init__.py +85 -0
- mlops_katib-0.19.0/kubeflow/katib/api/__init__.py +5 -0
- mlops_katib-0.19.0/kubeflow/katib/api/katib_client.py +1967 -0
- mlops_katib-0.19.0/kubeflow/katib/api/katib_client_test.py +1269 -0
- mlops_katib-0.19.0/kubeflow/katib/api/report_metrics.py +82 -0
- mlops_katib-0.19.0/kubeflow/katib/api/report_metrics_test.py +104 -0
- mlops_katib-0.19.0/kubeflow/katib/api/search.py +66 -0
- mlops_katib-0.19.0/kubeflow/katib/api_client.py +666 -0
- mlops_katib-0.19.0/kubeflow/katib/configuration.py +376 -0
- mlops_katib-0.19.0/kubeflow/katib/constants/__init__.py +0 -0
- mlops_katib-0.19.0/kubeflow/katib/constants/constants.py +83 -0
- mlops_katib-0.19.0/kubeflow/katib/exceptions.py +120 -0
- mlops_katib-0.19.0/kubeflow/katib/katib_api_pb2.py +133 -0
- mlops_katib-0.19.0/kubeflow/katib/katib_api_pb2_grpc.py +463 -0
- mlops_katib-0.19.0/kubeflow/katib/models/__init__.py +60 -0
- mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_algorithm_setting.py +150 -0
- mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_algorithm_spec.py +150 -0
- mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_collector_spec.py +146 -0
- mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_config_map_source.py +178 -0
- mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_early_stopping_rule.py +206 -0
- mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_early_stopping_setting.py +150 -0
- mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_early_stopping_spec.py +150 -0
- mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_experiment.py +228 -0
- mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_experiment_condition.py +260 -0
- mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_experiment_list.py +203 -0
- mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_experiment_spec.py +390 -0
- mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_experiment_status.py +646 -0
- mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_feasible_space.py +224 -0
- mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_file_system_path.py +172 -0
- mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_filter_spec.py +122 -0
- mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_graph_config.py +172 -0
- mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_metric.py +198 -0
- mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_metric_strategy.py +146 -0
- mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_metrics_collector_spec.py +146 -0
- mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_nas_config.py +146 -0
- mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_objective_spec.py +234 -0
- mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_observation.py +122 -0
- mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_operation.py +146 -0
- mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_optimal_trial.py +176 -0
- mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_parameter_assignment.py +146 -0
- mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_parameter_spec.py +172 -0
- mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_source_spec.py +172 -0
- mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_suggestion.py +228 -0
- mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_suggestion_condition.py +260 -0
- mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_suggestion_list.py +203 -0
- mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_suggestion_spec.py +202 -0
- mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_suggestion_status.py +284 -0
- mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_trial.py +228 -0
- mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_trial_assignment.py +206 -0
- mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_trial_condition.py +260 -0
- mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_trial_list.py +203 -0
- mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_trial_parameter_spec.py +178 -0
- mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_trial_source.py +146 -0
- mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_trial_spec.py +396 -0
- mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_trial_status.py +226 -0
- mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_trial_template.py +314 -0
- mlops_katib-0.19.0/kubeflow/katib/rest.py +291 -0
- mlops_katib-0.19.0/kubeflow/katib/types/__init__.py +0 -0
- mlops_katib-0.19.0/kubeflow/katib/types/types.py +10 -0
- mlops_katib-0.19.0/kubeflow/katib/utils/__init__.py +0 -0
- mlops_katib-0.19.0/kubeflow/katib/utils/utils.py +665 -0
- mlops_katib-0.19.0/kubeflow/katib/utils/utils_test.py +113 -0
- mlops_katib-0.19.0/mlops/__init__.py +1 -0
- mlops_katib-0.19.0/mlops/katib/__init__.py +22 -0
- mlops_katib-0.19.0/mlops_katib.egg-info/PKG-INFO +49 -0
- mlops_katib-0.19.0/mlops_katib.egg-info/SOURCES.txt +74 -0
- mlops_katib-0.19.0/mlops_katib.egg-info/dependency_links.txt +1 -0
- mlops_katib-0.19.0/mlops_katib.egg-info/not-zip-safe +1 -0
- mlops_katib-0.19.0/mlops_katib.egg-info/requires.txt +14 -0
- mlops_katib-0.19.0/mlops_katib.egg-info/top_level.txt +2 -0
- mlops_katib-0.19.0/pyproject.toml +3 -0
- mlops_katib-0.19.0/setup.cfg +4 -0
- mlops_katib-0.19.0/setup.py +94 -0
|
@@ -0,0 +1,49 @@
|
|
|
1
|
+
Metadata-Version: 2.4
|
|
2
|
+
Name: mlops-katib
|
|
3
|
+
Version: 0.19.0
|
|
4
|
+
Summary: MLOps Katib Python SDK for APIVersion v1beta1
|
|
5
|
+
Home-page: https://github.com/mlops-katib/mlops-katib
|
|
6
|
+
Author: Kubeflow Authors
|
|
7
|
+
Author-email: premnath.vel@gmail.com
|
|
8
|
+
License: Apache License Version 2.0
|
|
9
|
+
Classifier: Intended Audience :: Developers
|
|
10
|
+
Classifier: Intended Audience :: Education
|
|
11
|
+
Classifier: Intended Audience :: Science/Research
|
|
12
|
+
Classifier: Programming Language :: Python :: 3
|
|
13
|
+
Classifier: Programming Language :: Python :: 3 :: Only
|
|
14
|
+
Classifier: Programming Language :: Python :: 3.8
|
|
15
|
+
Classifier: Programming Language :: Python :: 3.9
|
|
16
|
+
Classifier: Programming Language :: Python :: 3.10
|
|
17
|
+
Classifier: Programming Language :: Python :: 3.11
|
|
18
|
+
Classifier: License :: OSI Approved :: Apache Software License
|
|
19
|
+
Classifier: Operating System :: OS Independent
|
|
20
|
+
Classifier: Topic :: Scientific/Engineering
|
|
21
|
+
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
|
|
22
|
+
Classifier: Topic :: Software Development
|
|
23
|
+
Classifier: Topic :: Software Development :: Libraries
|
|
24
|
+
Classifier: Topic :: Software Development :: Libraries :: Python Modules
|
|
25
|
+
Description-Content-Type: text/plain
|
|
26
|
+
Requires-Dist: certifi>=14.05.14
|
|
27
|
+
Requires-Dist: six>=1.10
|
|
28
|
+
Requires-Dist: setuptools>=21.0.0
|
|
29
|
+
Requires-Dist: urllib3>=1.15.1
|
|
30
|
+
Requires-Dist: kubernetes>=27.2.0
|
|
31
|
+
Requires-Dist: grpcio>=1.64.1
|
|
32
|
+
Requires-Dist: protobuf<5,>=4.21.12
|
|
33
|
+
Requires-Dist: kubeflow-training==1.9.0
|
|
34
|
+
Provides-Extra: huggingface
|
|
35
|
+
Requires-Dist: kubeflow-training[huggingface]==1.9.0; extra == "huggingface"
|
|
36
|
+
Provides-Extra: model-registry
|
|
37
|
+
Requires-Dist: model-registry-client; extra == "model-registry"
|
|
38
|
+
Dynamic: author
|
|
39
|
+
Dynamic: author-email
|
|
40
|
+
Dynamic: classifier
|
|
41
|
+
Dynamic: description
|
|
42
|
+
Dynamic: description-content-type
|
|
43
|
+
Dynamic: home-page
|
|
44
|
+
Dynamic: license
|
|
45
|
+
Dynamic: provides-extra
|
|
46
|
+
Dynamic: requires-dist
|
|
47
|
+
Dynamic: summary
|
|
48
|
+
|
|
49
|
+
MLOps Katib Python SDK for APIVersion v1beta1
|
|
@@ -0,0 +1,125 @@
|
|
|
1
|
+
# Kubeflow Katib SDK
|
|
2
|
+
|
|
3
|
+
Python SDK for Kubeflow Katib
|
|
4
|
+
|
|
5
|
+
## Requirements.
|
|
6
|
+
|
|
7
|
+
Python >= 3.8
|
|
8
|
+
|
|
9
|
+
Katib Python SDK follows [Python release cycle](https://devguide.python.org/versions/#python-release-cycle)
|
|
10
|
+
for supported Python versions.
|
|
11
|
+
|
|
12
|
+
## Installation & Usage
|
|
13
|
+
|
|
14
|
+
### pip install
|
|
15
|
+
|
|
16
|
+
```sh
|
|
17
|
+
pip install kubeflow-katib
|
|
18
|
+
```
|
|
19
|
+
|
|
20
|
+
Then import package:
|
|
21
|
+
|
|
22
|
+
```python
|
|
23
|
+
from kubeflow import katib
|
|
24
|
+
```
|
|
25
|
+
|
|
26
|
+
### Setuptools
|
|
27
|
+
|
|
28
|
+
Install via [Setuptools](http://pypi.python.org/pypi/setuptools).
|
|
29
|
+
|
|
30
|
+
```sh
|
|
31
|
+
python setup.py install --user
|
|
32
|
+
```
|
|
33
|
+
|
|
34
|
+
(or `sudo python setup.py install` to install the package for all users)
|
|
35
|
+
|
|
36
|
+
### Publish new SDK version to PyPi
|
|
37
|
+
|
|
38
|
+
Our SDK is located in [`kubeflow-katib` PyPi package](https://pypi.org/project/kubeflow-katib/).
|
|
39
|
+
Katib Python SDK is published as part of the Katib patch releases.
|
|
40
|
+
You can check the release process [here](../../../scripts/v1beta1/release.sh).
|
|
41
|
+
For each Katib patch release, we upload a new SDK version to the PyPi.
|
|
42
|
+
The SDK version is equal to the Katib version.
|
|
43
|
+
|
|
44
|
+
## Getting Started
|
|
45
|
+
|
|
46
|
+
Please follow the [examples](../../../examples/v1beta1/sdk) to learn more about Katib SDK.
|
|
47
|
+
|
|
48
|
+
## Documentation for API Endpoints
|
|
49
|
+
|
|
50
|
+
| Class | Method | Description |
|
|
51
|
+
| --------------------- | -------------------------------------- | ----------------------------------------------------------- |
|
|
52
|
+
| [KatibClient][client] | [create_experiment][create] | Create the Katib Experiment |
|
|
53
|
+
| [KatibClient][client] | [get_experiment][get_e] | Get the Katib Experiment |
|
|
54
|
+
| [KatibClient][client] | [get_suggestion][get_s] | Get the Katib Suggestion |
|
|
55
|
+
| [KatibClient][client] | [delete_experiment][delete] | Delete the Katib Experiment |
|
|
56
|
+
| [KatibClient][client] | [list_experiments][list_e] | List all Katib Experiments |
|
|
57
|
+
| [KatibClient][client] | [get_experiment_status][get_status] | Get the Experiment current status |
|
|
58
|
+
| [KatibClient][client] | [is_experiment_succeeded][is_suc] | Check if Experiment has succeeded |
|
|
59
|
+
| [KatibClient][client] | [list_trials][list_t] | List all Experiment's Trials |
|
|
60
|
+
| [KatibClient][client] | [get_success_trial_details][get_suc_t] | Get the Trial details that have succeeded for an Experiment |
|
|
61
|
+
| [KatibClient][client] | [get_optimal_hyperparameters][opt_hp] | Get the current optimal Trial from the Experiment |
|
|
62
|
+
|
|
63
|
+
[client]: docs/KatibClient.md
|
|
64
|
+
[create]: docs/KatibClient.md#create_experiment
|
|
65
|
+
[get_e]: docs/KatibClient.md#get_experiment
|
|
66
|
+
[get_s]: docs/KatibClient.md#get_suggestion
|
|
67
|
+
[delete]: docs/KatibClient.md#delete_experiment
|
|
68
|
+
[list_e]: docs/KatibClient.md#list_experiments
|
|
69
|
+
[get_status]: docs/KatibClient.md#get_experiment_status
|
|
70
|
+
[is_suc]: docs/KatibClient.md#is_experiment_succeeded
|
|
71
|
+
[list_t]: docs/KatibClient.md#list_trials
|
|
72
|
+
[get_suc_t]: docs/KatibClient.md#get_success_trial_details
|
|
73
|
+
[opt_hp]: docs/KatibClient.md#get_optimal_hyperparameters
|
|
74
|
+
|
|
75
|
+
## Documentation For Models
|
|
76
|
+
|
|
77
|
+
- [V1beta1AlgorithmSetting](docs/V1beta1AlgorithmSetting.md)
|
|
78
|
+
- [V1beta1AlgorithmSpec](docs/V1beta1AlgorithmSpec.md)
|
|
79
|
+
- [V1beta1CollectorSpec](docs/V1beta1CollectorSpec.md)
|
|
80
|
+
- [V1beta1ConfigMapSource](docs/V1beta1ConfigMapSource.md)
|
|
81
|
+
- [V1beta1EarlyStoppingRule](docs/V1beta1EarlyStoppingRule.md)
|
|
82
|
+
- [V1beta1EarlyStoppingSetting](docs/V1beta1EarlyStoppingSetting.md)
|
|
83
|
+
- [V1beta1EarlyStoppingSpec](docs/V1beta1EarlyStoppingSpec.md)
|
|
84
|
+
- [V1beta1Experiment](docs/V1beta1Experiment.md)
|
|
85
|
+
- [V1beta1ExperimentCondition](docs/V1beta1ExperimentCondition.md)
|
|
86
|
+
- [V1beta1ExperimentList](docs/V1beta1ExperimentList.md)
|
|
87
|
+
- [V1beta1ExperimentSpec](docs/V1beta1ExperimentSpec.md)
|
|
88
|
+
- [V1beta1ExperimentStatus](docs/V1beta1ExperimentStatus.md)
|
|
89
|
+
- [V1beta1FeasibleSpace](docs/V1beta1FeasibleSpace.md)
|
|
90
|
+
- [V1beta1FileSystemPath](docs/V1beta1FileSystemPath.md)
|
|
91
|
+
- [V1beta1FilterSpec](docs/V1beta1FilterSpec.md)
|
|
92
|
+
- [V1beta1GraphConfig](docs/V1beta1GraphConfig.md)
|
|
93
|
+
- [V1beta1Metric](docs/V1beta1Metric.md)
|
|
94
|
+
- [V1beta1MetricStrategy](docs/V1beta1MetricStrategy.md)
|
|
95
|
+
- [V1beta1MetricsCollectorSpec](docs/V1beta1MetricsCollectorSpec.md)
|
|
96
|
+
- [V1beta1NasConfig](docs/V1beta1NasConfig.md)
|
|
97
|
+
- [V1beta1ObjectiveSpec](docs/V1beta1ObjectiveSpec.md)
|
|
98
|
+
- [V1beta1Observation](docs/V1beta1Observation.md)
|
|
99
|
+
- [V1beta1Operation](docs/V1beta1Operation.md)
|
|
100
|
+
- [V1beta1OptimalTrial](docs/V1beta1OptimalTrial.md)
|
|
101
|
+
- [V1beta1ParameterAssignment](docs/V1beta1ParameterAssignment.md)
|
|
102
|
+
- [V1beta1ParameterSpec](docs/V1beta1ParameterSpec.md)
|
|
103
|
+
- [V1beta1SourceSpec](docs/V1beta1SourceSpec.md)
|
|
104
|
+
- [V1beta1Suggestion](docs/V1beta1Suggestion.md)
|
|
105
|
+
- [V1beta1SuggestionCondition](docs/V1beta1SuggestionCondition.md)
|
|
106
|
+
- [V1beta1SuggestionList](docs/V1beta1SuggestionList.md)
|
|
107
|
+
- [V1beta1SuggestionSpec](docs/V1beta1SuggestionSpec.md)
|
|
108
|
+
- [V1beta1SuggestionStatus](docs/V1beta1SuggestionStatus.md)
|
|
109
|
+
- [V1beta1Trial](docs/V1beta1Trial.md)
|
|
110
|
+
- [V1beta1TrialAssignment](docs/V1beta1TrialAssignment.md)
|
|
111
|
+
- [V1beta1TrialCondition](docs/V1beta1TrialCondition.md)
|
|
112
|
+
- [V1beta1TrialList](docs/V1beta1TrialList.md)
|
|
113
|
+
- [V1beta1TrialParameterSpec](docs/V1beta1TrialParameterSpec.md)
|
|
114
|
+
- [V1beta1TrialSource](docs/V1beta1TrialSource.md)
|
|
115
|
+
- [V1beta1TrialSpec](docs/V1beta1TrialSpec.md)
|
|
116
|
+
- [V1beta1TrialStatus](docs/V1beta1TrialStatus.md)
|
|
117
|
+
- [V1beta1TrialTemplate](docs/V1beta1TrialTemplate.md)
|
|
118
|
+
|
|
119
|
+
## Documentation For Authorization
|
|
120
|
+
|
|
121
|
+
All endpoints do not require authorization.
|
|
122
|
+
|
|
123
|
+
## Author
|
|
124
|
+
|
|
125
|
+
prem0912
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
__path__ = __import__("pkgutil").extend_path(__path__, __name__)
|
|
@@ -0,0 +1,85 @@
|
|
|
1
|
+
# coding: utf-8
|
|
2
|
+
|
|
3
|
+
# flake8: noqa
|
|
4
|
+
|
|
5
|
+
"""
|
|
6
|
+
Katib
|
|
7
|
+
|
|
8
|
+
Swagger description for Katib # noqa: E501
|
|
9
|
+
|
|
10
|
+
The version of the OpenAPI document: v1beta1-0.1
|
|
11
|
+
Generated by: https://openapi-generator.tech
|
|
12
|
+
"""
|
|
13
|
+
|
|
14
|
+
|
|
15
|
+
from __future__ import absolute_import
|
|
16
|
+
|
|
17
|
+
__version__ = "0.1"
|
|
18
|
+
|
|
19
|
+
# import apis into sdk package
|
|
20
|
+
|
|
21
|
+
# import ApiClient
|
|
22
|
+
from kubeflow.katib.api_client import ApiClient
|
|
23
|
+
from kubeflow.katib.configuration import Configuration
|
|
24
|
+
from kubeflow.katib.exceptions import OpenApiException
|
|
25
|
+
from kubeflow.katib.exceptions import ApiTypeError
|
|
26
|
+
from kubeflow.katib.exceptions import ApiValueError
|
|
27
|
+
from kubeflow.katib.exceptions import ApiKeyError
|
|
28
|
+
from kubeflow.katib.exceptions import ApiException
|
|
29
|
+
# import models into sdk package
|
|
30
|
+
from kubeflow.katib.models.v1beta1_algorithm_setting import V1beta1AlgorithmSetting
|
|
31
|
+
from kubeflow.katib.models.v1beta1_algorithm_spec import V1beta1AlgorithmSpec
|
|
32
|
+
from kubeflow.katib.models.v1beta1_collector_spec import V1beta1CollectorSpec
|
|
33
|
+
from kubeflow.katib.models.v1beta1_config_map_source import V1beta1ConfigMapSource
|
|
34
|
+
from kubeflow.katib.models.v1beta1_early_stopping_rule import V1beta1EarlyStoppingRule
|
|
35
|
+
from kubeflow.katib.models.v1beta1_early_stopping_setting import V1beta1EarlyStoppingSetting
|
|
36
|
+
from kubeflow.katib.models.v1beta1_early_stopping_spec import V1beta1EarlyStoppingSpec
|
|
37
|
+
from kubeflow.katib.models.v1beta1_experiment import V1beta1Experiment
|
|
38
|
+
from kubeflow.katib.models.v1beta1_experiment_condition import V1beta1ExperimentCondition
|
|
39
|
+
from kubeflow.katib.models.v1beta1_experiment_list import V1beta1ExperimentList
|
|
40
|
+
from kubeflow.katib.models.v1beta1_experiment_spec import V1beta1ExperimentSpec
|
|
41
|
+
from kubeflow.katib.models.v1beta1_experiment_status import V1beta1ExperimentStatus
|
|
42
|
+
from kubeflow.katib.models.v1beta1_feasible_space import V1beta1FeasibleSpace
|
|
43
|
+
from kubeflow.katib.models.v1beta1_file_system_path import V1beta1FileSystemPath
|
|
44
|
+
from kubeflow.katib.models.v1beta1_filter_spec import V1beta1FilterSpec
|
|
45
|
+
from kubeflow.katib.models.v1beta1_graph_config import V1beta1GraphConfig
|
|
46
|
+
from kubeflow.katib.models.v1beta1_metric import V1beta1Metric
|
|
47
|
+
from kubeflow.katib.models.v1beta1_metric_strategy import V1beta1MetricStrategy
|
|
48
|
+
from kubeflow.katib.models.v1beta1_metrics_collector_spec import V1beta1MetricsCollectorSpec
|
|
49
|
+
from kubeflow.katib.models.v1beta1_nas_config import V1beta1NasConfig
|
|
50
|
+
from kubeflow.katib.models.v1beta1_objective_spec import V1beta1ObjectiveSpec
|
|
51
|
+
from kubeflow.katib.models.v1beta1_observation import V1beta1Observation
|
|
52
|
+
from kubeflow.katib.models.v1beta1_operation import V1beta1Operation
|
|
53
|
+
from kubeflow.katib.models.v1beta1_optimal_trial import V1beta1OptimalTrial
|
|
54
|
+
from kubeflow.katib.models.v1beta1_parameter_assignment import V1beta1ParameterAssignment
|
|
55
|
+
from kubeflow.katib.models.v1beta1_parameter_spec import V1beta1ParameterSpec
|
|
56
|
+
from kubeflow.katib.models.v1beta1_source_spec import V1beta1SourceSpec
|
|
57
|
+
from kubeflow.katib.models.v1beta1_suggestion import V1beta1Suggestion
|
|
58
|
+
from kubeflow.katib.models.v1beta1_suggestion_condition import V1beta1SuggestionCondition
|
|
59
|
+
from kubeflow.katib.models.v1beta1_suggestion_list import V1beta1SuggestionList
|
|
60
|
+
from kubeflow.katib.models.v1beta1_suggestion_spec import V1beta1SuggestionSpec
|
|
61
|
+
from kubeflow.katib.models.v1beta1_suggestion_status import V1beta1SuggestionStatus
|
|
62
|
+
from kubeflow.katib.models.v1beta1_trial import V1beta1Trial
|
|
63
|
+
from kubeflow.katib.models.v1beta1_trial_assignment import V1beta1TrialAssignment
|
|
64
|
+
from kubeflow.katib.models.v1beta1_trial_condition import V1beta1TrialCondition
|
|
65
|
+
from kubeflow.katib.models.v1beta1_trial_list import V1beta1TrialList
|
|
66
|
+
from kubeflow.katib.models.v1beta1_trial_parameter_spec import V1beta1TrialParameterSpec
|
|
67
|
+
from kubeflow.katib.models.v1beta1_trial_source import V1beta1TrialSource
|
|
68
|
+
from kubeflow.katib.models.v1beta1_trial_spec import V1beta1TrialSpec
|
|
69
|
+
from kubeflow.katib.models.v1beta1_trial_status import V1beta1TrialStatus
|
|
70
|
+
from kubeflow.katib.models.v1beta1_trial_template import V1beta1TrialTemplate
|
|
71
|
+
|
|
72
|
+
# Import Katib API client.
|
|
73
|
+
from kubeflow.katib.api.katib_client import KatibClient
|
|
74
|
+
from kubeflow.katib.api.katib_client import ModelRegistryArtifactType
|
|
75
|
+
from kubeflow.katib.api.katib_client import ModelRegistryConfig
|
|
76
|
+
# Import Katib TrainerResources class.
|
|
77
|
+
from kubeflow.katib.types.types import TrainerResources
|
|
78
|
+
# Import Katib report metrics functions
|
|
79
|
+
from kubeflow.katib.api.report_metrics import report_metrics
|
|
80
|
+
# Import Katib helper functions.
|
|
81
|
+
import kubeflow.katib.api.search as search
|
|
82
|
+
# Import Katib helper constants.
|
|
83
|
+
from kubeflow.katib.constants.constants import BASE_IMAGE_TENSORFLOW
|
|
84
|
+
from kubeflow.katib.constants.constants import BASE_IMAGE_TENSORFLOW_GPU
|
|
85
|
+
from kubeflow.katib.constants.constants import BASE_IMAGE_PYTORCH
|