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.
Files changed (76) hide show
  1. mlops_katib-0.19.0/PKG-INFO +49 -0
  2. mlops_katib-0.19.0/README.md +125 -0
  3. mlops_katib-0.19.0/kubeflow/__init__.py +1 -0
  4. mlops_katib-0.19.0/kubeflow/katib/__init__.py +85 -0
  5. mlops_katib-0.19.0/kubeflow/katib/api/__init__.py +5 -0
  6. mlops_katib-0.19.0/kubeflow/katib/api/katib_client.py +1967 -0
  7. mlops_katib-0.19.0/kubeflow/katib/api/katib_client_test.py +1269 -0
  8. mlops_katib-0.19.0/kubeflow/katib/api/report_metrics.py +82 -0
  9. mlops_katib-0.19.0/kubeflow/katib/api/report_metrics_test.py +104 -0
  10. mlops_katib-0.19.0/kubeflow/katib/api/search.py +66 -0
  11. mlops_katib-0.19.0/kubeflow/katib/api_client.py +666 -0
  12. mlops_katib-0.19.0/kubeflow/katib/configuration.py +376 -0
  13. mlops_katib-0.19.0/kubeflow/katib/constants/__init__.py +0 -0
  14. mlops_katib-0.19.0/kubeflow/katib/constants/constants.py +83 -0
  15. mlops_katib-0.19.0/kubeflow/katib/exceptions.py +120 -0
  16. mlops_katib-0.19.0/kubeflow/katib/katib_api_pb2.py +133 -0
  17. mlops_katib-0.19.0/kubeflow/katib/katib_api_pb2_grpc.py +463 -0
  18. mlops_katib-0.19.0/kubeflow/katib/models/__init__.py +60 -0
  19. mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_algorithm_setting.py +150 -0
  20. mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_algorithm_spec.py +150 -0
  21. mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_collector_spec.py +146 -0
  22. mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_config_map_source.py +178 -0
  23. mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_early_stopping_rule.py +206 -0
  24. mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_early_stopping_setting.py +150 -0
  25. mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_early_stopping_spec.py +150 -0
  26. mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_experiment.py +228 -0
  27. mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_experiment_condition.py +260 -0
  28. mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_experiment_list.py +203 -0
  29. mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_experiment_spec.py +390 -0
  30. mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_experiment_status.py +646 -0
  31. mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_feasible_space.py +224 -0
  32. mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_file_system_path.py +172 -0
  33. mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_filter_spec.py +122 -0
  34. mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_graph_config.py +172 -0
  35. mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_metric.py +198 -0
  36. mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_metric_strategy.py +146 -0
  37. mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_metrics_collector_spec.py +146 -0
  38. mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_nas_config.py +146 -0
  39. mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_objective_spec.py +234 -0
  40. mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_observation.py +122 -0
  41. mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_operation.py +146 -0
  42. mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_optimal_trial.py +176 -0
  43. mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_parameter_assignment.py +146 -0
  44. mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_parameter_spec.py +172 -0
  45. mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_source_spec.py +172 -0
  46. mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_suggestion.py +228 -0
  47. mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_suggestion_condition.py +260 -0
  48. mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_suggestion_list.py +203 -0
  49. mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_suggestion_spec.py +202 -0
  50. mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_suggestion_status.py +284 -0
  51. mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_trial.py +228 -0
  52. mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_trial_assignment.py +206 -0
  53. mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_trial_condition.py +260 -0
  54. mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_trial_list.py +203 -0
  55. mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_trial_parameter_spec.py +178 -0
  56. mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_trial_source.py +146 -0
  57. mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_trial_spec.py +396 -0
  58. mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_trial_status.py +226 -0
  59. mlops_katib-0.19.0/kubeflow/katib/models/v1beta1_trial_template.py +314 -0
  60. mlops_katib-0.19.0/kubeflow/katib/rest.py +291 -0
  61. mlops_katib-0.19.0/kubeflow/katib/types/__init__.py +0 -0
  62. mlops_katib-0.19.0/kubeflow/katib/types/types.py +10 -0
  63. mlops_katib-0.19.0/kubeflow/katib/utils/__init__.py +0 -0
  64. mlops_katib-0.19.0/kubeflow/katib/utils/utils.py +665 -0
  65. mlops_katib-0.19.0/kubeflow/katib/utils/utils_test.py +113 -0
  66. mlops_katib-0.19.0/mlops/__init__.py +1 -0
  67. mlops_katib-0.19.0/mlops/katib/__init__.py +22 -0
  68. mlops_katib-0.19.0/mlops_katib.egg-info/PKG-INFO +49 -0
  69. mlops_katib-0.19.0/mlops_katib.egg-info/SOURCES.txt +74 -0
  70. mlops_katib-0.19.0/mlops_katib.egg-info/dependency_links.txt +1 -0
  71. mlops_katib-0.19.0/mlops_katib.egg-info/not-zip-safe +1 -0
  72. mlops_katib-0.19.0/mlops_katib.egg-info/requires.txt +14 -0
  73. mlops_katib-0.19.0/mlops_katib.egg-info/top_level.txt +2 -0
  74. mlops_katib-0.19.0/pyproject.toml +3 -0
  75. mlops_katib-0.19.0/setup.cfg +4 -0
  76. 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
@@ -0,0 +1,5 @@
1
+ from __future__ import absolute_import
2
+
3
+ # flake8: noqa
4
+
5
+ # import apis into api package