Skip to content

Commit 7624db0

Browse files
Captainiabenieric
authored andcommitted
Merge master into benchmark feature (aws#4652)
1 parent f90b99f commit 7624db0

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

src/sagemaker/jumpstart/types.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -1097,7 +1097,7 @@ def __init__(
10971097
config (Dict[str, Any]):
10981098
Dictionary representation of the config.
10991099
base_fields (Dict[str, Any]):
1100-
The default base fields that are used to construct the final resolved config.
1100+
The default base fields that are used to construct the resolved config.
11011101
config_components (Dict[str, JumpStartConfigComponent]):
11021102
The list of components that are used to construct the resolved config.
11031103
"""

tests/unit/sagemaker/jumpstart/test_notebook_utils.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -241,15 +241,15 @@ def test_list_jumpstart_models_script_filter(
241241
manifest_length = len(get_prototype_manifest())
242242
vals = [True, False]
243243
for val in vals:
244-
kwargs = {"filter": f"training_supported == {val}"}
244+
kwargs = {"filter": And(f"training_supported == {val}", "model_type is open_weights")}
245245
list_jumpstart_models(**kwargs)
246246
assert patched_read_s3_file.call_count == 2 * manifest_length
247247
assert patched_get_manifest.call_count == 2
248248

249249
patched_get_manifest.reset_mock()
250250
patched_read_s3_file.reset_mock()
251251

252-
kwargs = {"filter": f"training_supported != {val}"}
252+
kwargs = {"filter": And(f"training_supported != {val}", "model_type is open_weights")}
253253
list_jumpstart_models(**kwargs)
254254
assert patched_read_s3_file.call_count == 2 * manifest_length
255255
assert patched_get_manifest.call_count == 2

0 commit comments

Comments
 (0)