Skip to content

Commit 7f042e9

Browse files
committed
chore: add deprecated function to ensure backwards compatibility
1 parent 17dd24e commit 7f042e9

File tree

1 file changed

+20
-0
lines changed

1 file changed

+20
-0
lines changed

src/sagemaker/jumpstart/accessors.py

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,8 @@
1313
"""This module contains accessors related to SageMaker JumpStart."""
1414
from __future__ import absolute_import
1515
from typing import Any, Dict, List, Optional
16+
17+
from sagemaker.deprecations import deprecated
1618
from sagemaker.jumpstart.types import JumpStartModelHeader, JumpStartModelSpecs
1719
from sagemaker.jumpstart import cache
1820
from sagemaker.jumpstart.constants import JUMPSTART_DEFAULT_REGION_NAME
@@ -166,3 +168,21 @@ def reset_cache(cache_kwargs: Dict[str, Any] = None, region: Optional[str] = Non
166168
"""
167169
cache_kwargs_dict = {} if cache_kwargs is None else cache_kwargs
168170
JumpStartModelsAccessor.set_cache_kwargs(cache_kwargs_dict, region)
171+
172+
@staticmethod
173+
@deprecated()
174+
def get_manifest(
175+
cache_kwargs: Optional[Dict[str, Any]] = None, region: Optional[str] = None
176+
) -> List[JumpStartModelHeader]:
177+
"""Return entire JumpStart models manifest.
178+
Raises:
179+
ValueError: If region in `cache_kwargs` is inconsistent with `region` argument.
180+
Args:
181+
cache_kwargs (Dict[str, Any]): Optional. Cache kwargs to use.
182+
(Default: None).
183+
region (str): Optional. The region to use for the cache.
184+
(Default: None).
185+
"""
186+
cache_kwargs_dict: Dict[str, Any] = {} if cache_kwargs is None else cache_kwargs
187+
JumpStartModelsAccessor.set_cache_kwargs(cache_kwargs_dict, region)
188+
return JumpStartModelsAccessor._cache.get_manifest() # type: ignore

0 commit comments

Comments
 (0)