oracle-ads 2.13.10rc0__py3-none-any.whl → 2.13.12__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 (31) hide show
  1. ads/aqua/app.py +13 -7
  2. ads/aqua/cli.py +15 -0
  3. ads/aqua/common/entities.py +31 -5
  4. ads/aqua/common/utils.py +35 -0
  5. ads/aqua/config/container_config.py +0 -1
  6. ads/aqua/evaluation/evaluation.py +5 -4
  7. ads/aqua/extension/deployment_handler.py +4 -1
  8. ads/aqua/extension/model_handler.py +1 -1
  9. ads/aqua/model/enums.py +19 -1
  10. ads/aqua/model/model.py +45 -36
  11. ads/aqua/model/utils.py +1 -2
  12. ads/aqua/modeldeployment/config_loader.py +815 -0
  13. ads/aqua/modeldeployment/constants.py +4 -1
  14. ads/aqua/modeldeployment/deployment.py +100 -124
  15. ads/aqua/modeldeployment/entities.py +4 -178
  16. ads/aqua/modeldeployment/model_group_config.py +240 -0
  17. ads/aqua/modeldeployment/utils.py +0 -539
  18. ads/common/work_request.py +39 -38
  19. ads/jobs/builders/infrastructure/dsc_job.py +121 -24
  20. ads/jobs/builders/infrastructure/dsc_job_runtime.py +71 -24
  21. ads/jobs/builders/runtimes/base.py +7 -5
  22. ads/jobs/builders/runtimes/pytorch_runtime.py +6 -8
  23. ads/jobs/templates/driver_pytorch.py +486 -172
  24. ads/jobs/templates/driver_utils.py +27 -11
  25. ads/model/service/oci_datascience_model_deployment.py +6 -11
  26. ads/telemetry/client.py +4 -4
  27. {oracle_ads-2.13.10rc0.dist-info → oracle_ads-2.13.12.dist-info}/METADATA +2 -2
  28. {oracle_ads-2.13.10rc0.dist-info → oracle_ads-2.13.12.dist-info}/RECORD +31 -29
  29. {oracle_ads-2.13.10rc0.dist-info → oracle_ads-2.13.12.dist-info}/WHEEL +0 -0
  30. {oracle_ads-2.13.10rc0.dist-info → oracle_ads-2.13.12.dist-info}/entry_points.txt +0 -0
  31. {oracle_ads-2.13.10rc0.dist-info → oracle_ads-2.13.12.dist-info}/licenses/LICENSE.txt +0 -0
ads/jobs/builders/runtimes/pytorch_runtime.py CHANGED
@@ -1,19 +1,19 @@
1
1
  #!/usr/bin/env python
2
- # -*- coding: utf-8; -*-
3
2
 
4
- # Copyright (c) 2023, 2024 Oracle and/or its affiliates.
3
+ # Copyright (c) 2023, 2025 Oracle and/or its affiliates.
5
4
  # Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
6
5
 
7
- from ads.jobs.builders.runtimes.artifact import PythonArtifact, GitPythonArtifact
6
+ from ads.jobs.builders.runtimes.artifact import GitPythonArtifact, PythonArtifact
8
7
  from ads.jobs.builders.runtimes.base import MultiNodeRuntime
9
8
  from ads.jobs.builders.runtimes.python_runtime import (
10
- PythonRuntime,
11
9
  GitPythonRuntime,
10
+ PythonRuntime,
12
11
  )
13
12
 
14
13
 
15
14
  class PyTorchDistributedRuntime(PythonRuntime, MultiNodeRuntime):
16
15
  """Represents runtime supporting PyTorch Distributed training."""
16
+
17
17
  CONST_GIT = "git"
18
18
  CONST_INPUT = "inputs"
19
19
  CONST_DEP = "dependencies"
@@ -169,13 +169,11 @@ class PyTorchDistributedRuntime(PythonRuntime, MultiNodeRuntime):
169
169
  def command(self):
170
170
  """The command for launching the workload."""
171
171
  return self.get_spec(self.CONST_COMMAND)
172
-
172
+
173
173
  @property
174
174
  def use_deepspeed(self):
175
175
  """Indicate whether whether to configure deepspeed for multi-node workload"""
176
- if self.get_spec(self.CONST_DEEPSPEED):
177
- return True
178
- return False
176
+ return bool(self.get_spec(self.CONST_DEEPSPEED))
179
177
 
180
178
 
181
179
  class PyTorchDistributedArtifact(PythonArtifact):