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