Update naming

DIAGNijmegen · Dec 8, 2023 · dfe5a2a · dfe5a2a
1 parent ebd89f0
commit dfe5a2a
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 2 deletions.
diff --git a/sagemaker_shim/utils.py → sagemaker_shim/extract.py b/sagemaker_shim/utils.py → sagemaker_shim/extract.py
diff --git 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 → tests/test_extract.py b/tests/test_utils.py → 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):