Release version 2.5.8

Author: TeamCity Agent User

.gitignore

@@ -0,0 +1,158 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+env/
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+.hypothesis/
+.pytest_cache
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+docs/build/
+docs/dask-worker-space/
+
+# PyBuilder
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# pyenv
+.python-version
+
+# celery beat schedule file
+celerybeat-schedule
+
+# SageMath parsed files
+*.sage.py
+
+# dotenv
+.env
+
+# virtualenv
+.venv
+venv/
+ENV/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+
+### Linux ###
+*~
+
+# IntelliJ/Pycharm settings
+*.iml
+.idea
+.vscode/
+
+# Mac OS metadata
+.DS_Store
+.AppleDouble
+.LSOverride
+
+# dask related
+dask-worker-space
+
+#sqllite databases
+*.db
+
+Untitled*.ipynb
+*.lock
+*.dirlock
+
+
+# test results
+testresults.xml
+
+# this is a separate repo
+notebooks/stable
+
+# this is a repo for http certs
+tests/sparta-pki
+
+docs/ads-latest.zip
+
+
+# html reports of covered code by pytests and
+# execution data collected by coverage.py
+*coverage_html_report*
+.coverage
+.coverage*
+**.sqlite
+
+extensions/**/node_modules/
+extensions/**/*.tgz
+logs/
+**/node_modules
+
+# side-effects of running notebooks
+**/.mlx_static/**
+
+# vim
+*.swp

ads/ads_version.json

@@ -1,3 +1,3 @@
 {
-  "version": "2.5.7"
+  "version": "2.5.8"
 }

ads/cli.py

@@ -6,6 +6,7 @@
 
 import click
 import ads.opctl.cli
+import ads.jobs.cli
 import os
 import json
 
@@ -23,6 +24,7 @@ def cli():
 
 
 cli.add_command(ads.opctl.cli.commands)
+cli.add_command(ads.jobs.cli.commands)
 
 
 if __name__ == "__main__":

ads/common/model_artifact.py

@@ -16,7 +16,7 @@
 import uuid
 from enum import Enum
 from pathlib import Path
-from typing import Union
+from typing import Union, Any
 from urllib.parse import urlparse
 import fsspec
 import numpy as np
@@ -1258,11 +1258,6 @@ def _populate_metadata_taxonomy(self, model=None, use_case_type=None):
     def _populate_metadata_custom(self):
         """Extracts custom metadata info from model artifact.
 
-        Parameters
-        ----------
-        None
-            Nothing
-
         Returns
         -------
         None
@@ -1349,16 +1344,17 @@ def _populate_metadata_custom(self):
         )
         self.metadata_custom._add_many(model_metadata_items, replace=True)
 
-    def populate_metadata(self, model=None, use_case_type=None):
+    def populate_metadata(self, model: Any = None, use_case_type: str = None):
         """Extracts and populate taxonomy metadata from given model.
 
         Parameters
         ----------
-        model: [sklearn, xgboost, lightgbm, automl, keras]
-            The model object
-
-        use_case_type:
-            The use case type of the model
+        model: (Any, optional). Defaults to None.
+            This is an optional model object which is only used to extract taxonomy metadata.
+            Supported models: automl, keras, lightgbm, pytorch, sklearn, tensorflow, and xgboost.
+            If the model is not under supported frameworks, then extracting taxonomy metadata will be skipped.
+        use_case_type: (str, optional). Default to None.
+            The use case type of the model.
 
         Returns
         -------
@@ -1367,7 +1363,8 @@ def populate_metadata(self, model=None, use_case_type=None):
         """
         if model is None and self.metadata_taxonomy["Algorithm"].value is None:
             logger.info(
-                "To auto-extract taxonomy metadata the model must be provided. Supported models: automl, keras, lightgbm, pytorch, sklearn, tensorflow, and xgboost."
+                "To auto-extract taxonomy metadata the model must be provided. "
+                "Supported models: automl, keras, lightgbm, pytorch, sklearn, tensorflow, and xgboost."
             )
         if use_case_type is None:
             use_case_type = self.metadata_taxonomy[