amzn-sagemaker-checkpointing 1.0.10__tar.gz → 1.0.13__tar.gz

amzn_sagemaker_checkpointing-1.0.13/DEVELOPING.md

@@ -0,0 +1,22 @@
+# Developing SageMakerCheckpointing
+
+This package uses the [hatch](https://hatch.pypa.io/latest/) build system.
+
+### Building
+
+A number of scripts and commands exist in `pyproject.toml` under the `scripts` configurations with more
+documentation in the comments of `pyproject.toml`. Running a script for a specific environment is simply running
+`hatch run <env_name>:<script>`. You can omit the `<env_name>` for those under the `default` environment.
+
+You need to set up hatch pluging first:
+```
+./setup-hatch.sh
+```
+
+### Available Hatch Commands
+
+- **`hatch run release`** - Runs typing checks (mypy), tests, and coverage.
+- **`hatch test --cover`** - Runs tests and coverage.
+- **`hatch typing`** - Runs mypy type checking.
+- **`hatch fmt`** - Formats code using ruff.
+- **`hatch build`** - builds both source and wheel distributions in ./build directory.

{amzn_sagemaker_checkpointing-1.0.10 → amzn_sagemaker_checkpointing-1.0.13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: amzn-sagemaker-checkpointing
-Version: 1.0.10
+Version: 1.0.13
 Summary: Amazon SageMaker Checkpointing Library
 License: Apache 2.0
 License-File: LICENSE.txt
@@ -95,12 +95,12 @@ following to your S3 bucket policy
 ```
 
 ## Installation
-### PreRequisites
+### Prerequisites
 ```bash
 pip install s3torchconnector tenacity torch boto3 botocore
 ```
 
-### Install amzn-sagemaker-checkpointing library
+### SageMaker Checkpointing Library
 ```bash
 pip install amzn-sagemaker-checkpointing
 ```

{amzn_sagemaker_checkpointing-1.0.10 → amzn_sagemaker_checkpointing-1.0.13}/README.md

@@ -82,12 +82,12 @@ following to your S3 bucket policy
 ```
 
 ## Installation
-### PreRequisites
+### Prerequisites
 ```bash
 pip install s3torchconnector tenacity torch boto3 botocore
 ```
 
-### Install amzn-sagemaker-checkpointing library
+### SageMaker Checkpointing Library
 ```bash
 pip install amzn-sagemaker-checkpointing
 ```

{amzn_sagemaker_checkpointing-1.0.10 → amzn_sagemaker_checkpointing-1.0.13}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "amzn-sagemaker-checkpointing"
-version = "1.0.10"
+version = "1.0.13"
 description = "Amazon SageMaker Checkpointing Library"
 readme = "README.md"
 license = { "text" = "Apache 2.0" }
@@ -62,9 +62,12 @@ output = "private/brazil-documentation/coverage/coverage.xml"
 [tool.coverage.html]
 directory = "private/brazil-documentation/coverage/"
 
+[tool.ruff]
+exclude = [ "./build", ".hatch", "private", ".venv" ]
+
 [tool.ruff.lint]
 isort.known-first-party = ["amzn_sagemaker_checkpointing"]
-exclude = [ "./build", ".hatch", "private" ]
+select = ["E", "W", "F", "I", "N", "RUF"]
 
 [tool.ruff.lint.per-file-ignores]
 # Tests can use magic values, assertions, and relative imports
@@ -72,6 +75,7 @@ exclude = [ "./build", ".hatch", "private" ]
 
 [tool.hatch.build]
 directory = "./build"
+exclude = ["DEVELOPING_INTERNAL.md"]
 
 [tool.hatch.env]
 requires = [ "hatch-pip-compile" ]
@@ -89,6 +93,7 @@ update = [ "hatch-pip-compile --upgrade --all" ]
 generate-version = [ "hatch version | tail -1 > ./build/latest_version" ]
 
 release = [
+    "hatch fmt",
     "typing",
     "hatch test --all --cover",
     "mkdir -p ./build",

amzn_sagemaker_checkpointing-1.0.13/setup-hatch.sh

@@ -0,0 +1,19 @@
+#!/bin/bash
+set -e
+
+mkdir -p .hatch
+
+cat > .hatch/hatch_plugin.py << 'EOF'
+from hatch.env.collectors.plugin.interface import EnvironmentCollectorInterface
+
+class CustomEnvironmentCollector(EnvironmentCollectorInterface):
+    PLUGIN_NAME = 'custom'
+    
+    def get_initial_config(self):
+        return {}
+    
+    def finalize_config(self, config):
+        return config
+EOF
+
+echo "Hatch plugin created"