ads.model.common package

Submodules

ads.model.common.utils module

class ads.model.common.utils.MetadataArtifactPathType

Bases: ExtendedEnum

Enum for defining metadata artifact path type. Can be either local path or OSS path. It can also be the content itself.

CONTENT = 'content'

LOCAL = 'local'

OSS = 'oss'

ads.model.common.utils.fetch_manifest_from_conda_location(env_location: str)

Convenience method to fetch the manifest file from a conda environment.

Parameters:

env_location (str) – Absolute path to the environment.

ads.model.common.utils.generate_model_artifact_manifest(artifact_dir: str, files) → Dict[str, Any]

Builds a manifest for files included in an ADS-generated artifact.

ads.model.common.utils.safe_extract_zip(zip_file: ZipFile, target_dir: str)

Extracts a model artifact archive after validating member paths.

ads.model.common.utils.verify_model_artifact_manifest(artifact_dir: str, strict: bool = False) → bool

Verifies artifact files when an ADS manifest is present.

ads.model.common.utils.zip_artifact(artifact_dir: str) → str

Prepares model artifacts ZIP archive.

Parameters:

artifact_dir (str) – Path to the model artifact.

Returns:

Path to the model artifact ZIP archive file.

Return type:

str

Module contents