From dfe5a2a25d63fe2dd7afde32b269c14bcb0044de Mon Sep 17 00:00:00 2001 From: James Meakin <12661555+jmsmkn@users.noreply.github.com> Date: Fri, 8 Dec 2023 13:08:19 +0100 Subject: [PATCH] Update naming --- sagemaker_shim/{utils.py => extract.py} | 0 sagemaker_shim/models.py | 2 +- tests/{test_utils.py => test_extract.py} | 2 +- 3 files changed, 2 insertions(+), 2 deletions(-) rename sagemaker_shim/{utils.py => extract.py} (100%) rename tests/{test_utils.py => test_extract.py} (98%) diff --git a/sagemaker_shim/utils.py b/sagemaker_shim/extract.py similarity index 100% rename from sagemaker_shim/utils.py rename to sagemaker_shim/extract.py diff --git a/sagemaker_shim/models.py b/sagemaker_shim/models.py index 30ccc79..532a76e 100644 --- a/sagemaker_shim/models.py +++ b/sagemaker_shim/models.py @@ -18,8 +18,8 @@ from pydantic import BaseModel, ConfigDict, field_validator from sagemaker_shim.exceptions import ZipExtractionError +from sagemaker_shim.extract import safe_extract from sagemaker_shim.logging import STDOUT_LEVEL -from sagemaker_shim.utils import safe_extract if TYPE_CHECKING: from mypy_boto3_s3 import S3Client # pragma: no cover diff --git a/tests/test_utils.py b/tests/test_extract.py similarity index 98% rename from tests/test_utils.py rename to tests/test_extract.py index 3f8a750..7c80d35 100644 --- a/tests/test_utils.py +++ b/tests/test_extract.py @@ -4,7 +4,7 @@ import pytest from sagemaker_shim.exceptions import ZipExtractionError -from sagemaker_shim.utils import safe_extract +from sagemaker_shim.extract import safe_extract def test_extract_with_dodgy_file(tmp_path):