mlops-katib 0.19.0__py3-none-any.whl
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.
- kubeflow/__init__.py +1 -0
- kubeflow/katib/__init__.py +85 -0
- kubeflow/katib/api/__init__.py +5 -0
- kubeflow/katib/api/katib_client.py +1967 -0
- kubeflow/katib/api/katib_client_test.py +1269 -0
- kubeflow/katib/api/report_metrics.py +82 -0
- kubeflow/katib/api/report_metrics_test.py +104 -0
- kubeflow/katib/api/search.py +66 -0
- kubeflow/katib/api_client.py +666 -0
- kubeflow/katib/configuration.py +376 -0
- kubeflow/katib/constants/__init__.py +0 -0
- kubeflow/katib/constants/constants.py +83 -0
- kubeflow/katib/exceptions.py +120 -0
- kubeflow/katib/katib_api_pb2.py +133 -0
- kubeflow/katib/katib_api_pb2_grpc.py +463 -0
- kubeflow/katib/models/__init__.py +60 -0
- kubeflow/katib/models/v1beta1_algorithm_setting.py +150 -0
- kubeflow/katib/models/v1beta1_algorithm_spec.py +150 -0
- kubeflow/katib/models/v1beta1_collector_spec.py +146 -0
- kubeflow/katib/models/v1beta1_config_map_source.py +178 -0
- kubeflow/katib/models/v1beta1_early_stopping_rule.py +206 -0
- kubeflow/katib/models/v1beta1_early_stopping_setting.py +150 -0
- kubeflow/katib/models/v1beta1_early_stopping_spec.py +150 -0
- kubeflow/katib/models/v1beta1_experiment.py +228 -0
- kubeflow/katib/models/v1beta1_experiment_condition.py +260 -0
- kubeflow/katib/models/v1beta1_experiment_list.py +203 -0
- kubeflow/katib/models/v1beta1_experiment_spec.py +390 -0
- kubeflow/katib/models/v1beta1_experiment_status.py +646 -0
- kubeflow/katib/models/v1beta1_feasible_space.py +224 -0
- kubeflow/katib/models/v1beta1_file_system_path.py +172 -0
- kubeflow/katib/models/v1beta1_filter_spec.py +122 -0
- kubeflow/katib/models/v1beta1_graph_config.py +172 -0
- kubeflow/katib/models/v1beta1_metric.py +198 -0
- kubeflow/katib/models/v1beta1_metric_strategy.py +146 -0
- kubeflow/katib/models/v1beta1_metrics_collector_spec.py +146 -0
- kubeflow/katib/models/v1beta1_nas_config.py +146 -0
- kubeflow/katib/models/v1beta1_objective_spec.py +234 -0
- kubeflow/katib/models/v1beta1_observation.py +122 -0
- kubeflow/katib/models/v1beta1_operation.py +146 -0
- kubeflow/katib/models/v1beta1_optimal_trial.py +176 -0
- kubeflow/katib/models/v1beta1_parameter_assignment.py +146 -0
- kubeflow/katib/models/v1beta1_parameter_spec.py +172 -0
- kubeflow/katib/models/v1beta1_source_spec.py +172 -0
- kubeflow/katib/models/v1beta1_suggestion.py +228 -0
- kubeflow/katib/models/v1beta1_suggestion_condition.py +260 -0
- kubeflow/katib/models/v1beta1_suggestion_list.py +203 -0
- kubeflow/katib/models/v1beta1_suggestion_spec.py +202 -0
- kubeflow/katib/models/v1beta1_suggestion_status.py +284 -0
- kubeflow/katib/models/v1beta1_trial.py +228 -0
- kubeflow/katib/models/v1beta1_trial_assignment.py +206 -0
- kubeflow/katib/models/v1beta1_trial_condition.py +260 -0
- kubeflow/katib/models/v1beta1_trial_list.py +203 -0
- kubeflow/katib/models/v1beta1_trial_parameter_spec.py +178 -0
- kubeflow/katib/models/v1beta1_trial_source.py +146 -0
- kubeflow/katib/models/v1beta1_trial_spec.py +396 -0
- kubeflow/katib/models/v1beta1_trial_status.py +226 -0
- kubeflow/katib/models/v1beta1_trial_template.py +314 -0
- kubeflow/katib/rest.py +291 -0
- kubeflow/katib/types/__init__.py +0 -0
- kubeflow/katib/types/types.py +10 -0
- kubeflow/katib/utils/__init__.py +0 -0
- kubeflow/katib/utils/utils.py +665 -0
- kubeflow/katib/utils/utils_test.py +113 -0
- mlops/__init__.py +1 -0
- mlops/katib/__init__.py +22 -0
- mlops_katib-0.19.0.dist-info/METADATA +49 -0
- mlops_katib-0.19.0.dist-info/RECORD +69 -0
- mlops_katib-0.19.0.dist-info/WHEEL +5 -0
- mlops_katib-0.19.0.dist-info/top_level.txt +2 -0
kubeflow/__init__.py
ADDED
|
@@ -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
|