oracle-ads 2.11.15__py3-none-any.whl → 2.11.17__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 (44) hide show
  1. ads/aqua/app.py +5 -6
  2. ads/aqua/common/entities.py +17 -0
  3. ads/aqua/common/enums.py +14 -1
  4. ads/aqua/common/utils.py +160 -3
  5. ads/aqua/config/config.py +1 -1
  6. ads/aqua/config/deployment_config_defaults.json +29 -1
  7. ads/aqua/config/resource_limit_names.json +1 -0
  8. ads/aqua/constants.py +6 -1
  9. ads/aqua/evaluation/entities.py +0 -1
  10. ads/aqua/evaluation/evaluation.py +47 -14
  11. ads/aqua/extension/common_handler.py +75 -5
  12. ads/aqua/extension/common_ws_msg_handler.py +57 -0
  13. ads/aqua/extension/deployment_handler.py +16 -13
  14. ads/aqua/extension/deployment_ws_msg_handler.py +54 -0
  15. ads/aqua/extension/errors.py +1 -1
  16. ads/aqua/extension/evaluation_ws_msg_handler.py +28 -6
  17. ads/aqua/extension/model_handler.py +134 -8
  18. ads/aqua/extension/models/ws_models.py +78 -3
  19. ads/aqua/extension/models_ws_msg_handler.py +49 -0
  20. ads/aqua/extension/ui_websocket_handler.py +7 -1
  21. ads/aqua/model/entities.py +28 -0
  22. ads/aqua/model/model.py +544 -129
  23. ads/aqua/modeldeployment/deployment.py +102 -43
  24. ads/aqua/modeldeployment/entities.py +9 -20
  25. ads/aqua/ui.py +152 -28
  26. ads/common/object_storage_details.py +2 -5
  27. ads/common/serializer.py +2 -3
  28. ads/jobs/builders/infrastructure/dsc_job.py +41 -12
  29. ads/jobs/builders/infrastructure/dsc_job_runtime.py +74 -27
  30. ads/jobs/builders/runtimes/container_runtime.py +83 -4
  31. ads/opctl/operator/lowcode/anomaly/const.py +1 -0
  32. ads/opctl/operator/lowcode/anomaly/model/base_model.py +23 -7
  33. ads/opctl/operator/lowcode/anomaly/operator_config.py +1 -0
  34. ads/opctl/operator/lowcode/anomaly/schema.yaml +4 -0
  35. ads/opctl/operator/lowcode/common/errors.py +6 -0
  36. ads/opctl/operator/lowcode/forecast/model/arima.py +3 -1
  37. ads/opctl/operator/lowcode/forecast/model/base_model.py +21 -13
  38. ads/opctl/operator/lowcode/forecast/model_evaluator.py +11 -2
  39. ads/pipeline/ads_pipeline_run.py +13 -2
  40. {oracle_ads-2.11.15.dist-info → oracle_ads-2.11.17.dist-info}/METADATA +2 -1
  41. {oracle_ads-2.11.15.dist-info → oracle_ads-2.11.17.dist-info}/RECORD +44 -40
  42. {oracle_ads-2.11.15.dist-info → oracle_ads-2.11.17.dist-info}/LICENSE.txt +0 -0
  43. {oracle_ads-2.11.15.dist-info → oracle_ads-2.11.17.dist-info}/WHEEL +0 -0
  44. {oracle_ads-2.11.15.dist-info → oracle_ads-2.11.17.dist-info}/entry_points.txt +0 -0
ads/aqua/model/entities.py CHANGED
@@ -14,6 +14,7 @@ from dataclasses import InitVar, dataclass, field
14
14
  from typing import List, Optional
15
15
 
16
16
  import oci
17
+ from huggingface_hub import hf_api
17
18
 
18
19
  from ads.aqua import logger
19
20
  from ads.aqua.app import CLIBuilderMixin
@@ -22,6 +23,7 @@ from ads.aqua.constants import LIFECYCLE_DETAILS_MISSING_JOBRUN, UNKNOWN_VALUE
22
23
  from ads.aqua.data import AquaResourceIdentifier
23
24
  from ads.aqua.model.enums import FineTuningDefinedMetadata
24
25
  from ads.aqua.training.exceptions import exit_code_dict
26
+ from ads.aqua.ui import ModelFormat
25
27
  from ads.common.serializer import DataClassSerializable
26
28
  from ads.common.utils import get_log_links
27
29
  from ads.model.datascience_model import DataScienceModel
@@ -41,6 +43,14 @@ class AquaFineTuneValidation(DataClassSerializable):
41
43
  value: str = ""
42
44
 
43
45
 
46
+ @dataclass(repr=False)
47
+ class ModelValidationResult:
48
+ model_file: Optional[str] = None
49
+ model_formats: List[ModelFormat] = field(default_factory=list)
50
+ telemetry_model_name: str = None
51
+ tags: Optional[dict] = None
52
+
53
+
44
54
  @dataclass(repr=False)
45
55
  class AquaFineTuningMetric(DataClassSerializable):
46
56
  name: str = field(default_factory=str)
@@ -76,6 +86,10 @@ class AquaModelSummary(DataClassSerializable):
76
86
  ready_to_deploy: bool = True
77
87
  ready_to_finetune: bool = False
78
88
  ready_to_import: bool = False
89
+ nvidia_gpu_supported: bool = False
90
+ arm_cpu_supported: bool = False
91
+ model_file: Optional[str] = None
92
+ model_formats: List[ModelFormat] = field(default_factory=list)
79
93
 
80
94
 
81
95
  @dataclass(repr=False)
@@ -86,6 +100,7 @@ class AquaModel(AquaModelSummary, DataClassSerializable):
86
100
  inference_container: str = None
87
101
  finetuning_container: str = None
88
102
  evaluation_container: str = None
103
+ artifact_location: str = None
89
104
 
90
105
 
91
106
  @dataclass(repr=False)
@@ -96,6 +111,16 @@ class HFModelContainerInfo:
96
111
  finetuning_container: str = None
97
112
 
98
113
 
114
+ @dataclass(repr=False)
115
+ class HFModelSummary:
116
+ """Represents a summary of Hugging Face model."""
117
+
118
+ model_info: hf_api.ModelInfo = field(default_factory=hf_api.ModelInfo)
119
+ aqua_model_info: Optional[AquaModelSummary] = field(
120
+ default_factory=AquaModelSummary
121
+ )
122
+
123
+
99
124
  @dataclass(repr=False)
100
125
  class AquaEvalFTCommon(DataClassSerializable):
101
126
  """Represents common fields for evaluation and fine-tuning."""
@@ -255,10 +280,13 @@ class AquaFineTuneModel(AquaModel, AquaEvalFTCommon, DataClassSerializable):
255
280
  class ImportModelDetails(CLIBuilderMixin):
256
281
  model: str
257
282
  os_path: str
283
+ download_from_hf: Optional[bool] = True
284
+ local_dir: Optional[str] = None
258
285
  inference_container: Optional[str] = None
259
286
  finetuning_container: Optional[str] = None
260
287
  compartment_id: Optional[str] = None
261
288
  project_id: Optional[str] = None
289
+ model_file: Optional[str] = None
262
290
 
263
291
  def __post_init__(self):
264
292
  self._command = "model register"