File tree 2 files changed +3
-3
lines changed
tests/unit/sagemaker/jumpstart
2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -1097,7 +1097,7 @@ def __init__(
1097
1097
config (Dict[str, Any]):
1098
1098
Dictionary representation of the config.
1099
1099
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.
1101
1101
config_components (Dict[str, JumpStartConfigComponent]):
1102
1102
The list of components that are used to construct the resolved config.
1103
1103
"""
Original file line number Diff line number Diff line change @@ -241,15 +241,15 @@ def test_list_jumpstart_models_script_filter(
241
241
manifest_length = len (get_prototype_manifest ())
242
242
vals = [True , False ]
243
243
for val in vals :
244
- kwargs = {"filter" : f"training_supported == { val } " }
244
+ kwargs = {"filter" : And ( f"training_supported == { val } " , "model_type is open_weights" ) }
245
245
list_jumpstart_models (** kwargs )
246
246
assert patched_read_s3_file .call_count == 2 * manifest_length
247
247
assert patched_get_manifest .call_count == 2
248
248
249
249
patched_get_manifest .reset_mock ()
250
250
patched_read_s3_file .reset_mock ()
251
251
252
- kwargs = {"filter" : f"training_supported != { val } " }
252
+ kwargs = {"filter" : And ( f"training_supported != { val } " , "model_type is open_weights" ) }
253
253
list_jumpstart_models (** kwargs )
254
254
assert patched_read_s3_file .call_count == 2 * manifest_length
255
255
assert patched_get_manifest .call_count == 2
You can’t perform that action at this time.
0 commit comments