File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed
tests/unit/sagemaker/jumpstart Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff 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 """
Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments