I believe this issue is very similar to #7898. I'm also using moto to test the retrieval of models, this time model packages instead of model package groups. I've created some model packages using create_model_package and when using the Resource ModelPackage in sagemaker_client.search, no model packages were returned. On the other hand, I had results returned when using the Resource ModelPackageGroup.
I also tried to list_model_packages with the same set of created model packages and had two outcomes:
When using ModelPackageGroupName: the models that were under the provided model package group were returned.
When not using any parameters: an empty list was returned.
As seen above, the sm_client.search did not return any results when using the ModelPackage Resource. The sm_client.list_model_packages method also did not return anything when passing no parameters, but it did when passing the ModelPackageGroupName parameter.
Environment Information
Python Version: 3.10.14 boto3 == 1.24.4 pytest == 8.3.2 moto == 5.0.13
Problem Description
I believe this issue is very similar to #7898. I'm also using moto to test the retrieval of models, this time model packages instead of model package groups. I've created some model packages using
create_model_package
and when using the ResourceModelPackage
insagemaker_client.search
, no model packages were returned. On the other hand, I had results returned when using the ResourceModelPackageGroup
.I also tried to
list_model_packages
with the same set of created model packages and had two outcomes:ModelPackageGroupName
: the models that were under the provided model package group were returned.Minimal Example
search_model_package.py
test_model_groups.py
When running
pytest -s test_model_packages.py
, the following print statements were shown in the console:tests/test_model_packages.py Sagemaker Search Function:
.List Model Package Function
.List Model Package Function With Package Group as Parameter
As seen above, the
sm_client.search
did not return any results when using theModelPackage
Resource. Thesm_client.list_model_packages
method also did not return anything when passing no parameters, but it did when passing theModelPackageGroupName
parameter.