Yasuke/moto
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Sagemaker: list_model_package(): Fixed a bug in CreationTime params (#6639)

Browse Source
This commit is contained in:
Bert Blommers 2023-08-12 21:12:49 +00:00 committed by GitHub
parent c8b5470e25
commit d47f984175
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
moto/sagemaker/models.py
View File

@ -3,6 +3,7 @@ import os
import random import random
import string import string
from datetime import datetime from datetime import datetime
from dateutil.tz import tzutc
from typing import Any, Dict, List, Optional, Iterable, Union from typing import Any, Dict, List, Optional, Iterable, Union
from moto.core import BaseBackend, BackendDict, BaseModel, CloudFormationModel from moto.core import BaseBackend, BackendDict, BaseModel, CloudFormationModel
@ -994,7 +995,7 @@ class ModelPackage(BaseObject):
_type="model-package", _type="model-package",
_id=model_package_name, _id=model_package_name,
) )
datetime_now = datetime.utcnow() datetime_now = datetime.now(tzutc())
self.model_package_name = model_package_name self.model_package_name = model_package_name
self.model_package_group_name = model_package_group_name self.model_package_group_name = model_package_group_name
self.model_package_version = model_package_version self.model_package_version = model_package_version
@ -2868,9 +2869,13 @@ class SageMakerModelBackend(BaseBackend):
sort_order: Optional[str], sort_order: Optional[str],
) -> List[ModelPackage]: ) -> List[ModelPackage]:
if isinstance(creation_time_before, int): if isinstance(creation_time_before, int):
creation_time_before_datetime = datetime.fromtimestamp(creation_time_before) creation_time_before_datetime = datetime.fromtimestamp(
creation_time_before, tz=tzutc()
)
if isinstance(creation_time_after, int): if isinstance(creation_time_after, int):
creation_time_after_datetime = datetime.fromtimestamp(creation_time_after) creation_time_after_datetime = datetime.fromtimestamp(
creation_time_after, tz=tzutc()
)
if model_package_group_name is not None: if model_package_group_name is not None:
model_package_type = "Versioned" model_package_type = "Versioned"
model_package_summary_list = list( model_package_summary_list = list(