capo-sagemaker 0.1.0__tar.gz

capo_sagemaker-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Karen Petrosyan
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

capo_sagemaker-0.1.0/PKG-INFO

@@ -0,0 +1,102 @@
+Metadata-Version: 2.4
+Name: capo-sagemaker
+Version: 0.1.0
+Summary: Python SDK for Sagemaker.
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
+Classifier: Topic :: Software Development :: Code Generators
+Classifier: Typing :: Typed
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: jmespath>=1.0
+Requires-Dist: zapros>=0.13.0
+Provides-Extra: sso
+Requires-Dist: capo-sso>=0.2.0; extra == "sso"
+Requires-Dist: capo-sso-oidc>=0.2.0; extra == "sso"
+Requires-Dist: capo-sts>=0.3.0; extra == "sso"
+Dynamic: license-file
+
+# Getting Started
+
+## Installation
+
+```
+pip install capo-sagemaker
+```
+
+## Usage
+
+```python
+from capo_sagemaker import AsyncSageMakerClient
+
+
+async def main():
+    async with AsyncSageMakerClient() as sage_maker:
+        # Example: call the add_association operation
+        response = await sage_maker.add_association()
+        print(response["source_arn"])
+```
+
+## Pagination
+
+Some operations in this SDK support pagination. If the operation supports pagination it will have an `iter_` prefixed method that returns an async iterator.
+
+```python
+from capo_sagemaker import AsyncSageMakerClient
+
+
+async def main():
+    async with AsyncSageMakerClient() as sage_maker:
+        # Example: paginate over create_hub_content_presigned_urls
+        async for item in sage_maker.iter_create_hub_content_presigned_urls():
+            print(item)
+```
+
+## Error Handling
+
+The SDK raises exceptions for errors returned by the API. Catch them to handle failures gracefully.
+
+```python
+from capo_sagemaker import AsyncSageMakerClient
+from capo_sagemaker.error import ResourceLimitExceeded
+
+
+async def main():
+    async with AsyncSageMakerClient() as sage_maker:
+        try:
+            await sage_maker.add_association()
+        except ResourceLimitExceeded as e:
+            print(f"Error: {e}")
+            print(e.data)  # additional error data
+```
+
+## Retrying
+
+The SDK retries failed operations automatically. Retry behaviour follows the Smithy specification: errors are retried based on their `is_retryable` and `is_throttling_error` attributes. Throttling errors use a longer base delay. Network-level failures (connection errors and timeouts) are also retried. Non-retryable errors, such as client errors without the `@retryable` trait, are raised immediately without further attempts.
+
+The number of attempts defaults to 3 and can be changed at the client level via `retry_max_attempts`, or per call via `config_overrides`.
+
+```python
+from capo_sagemaker import AsyncSageMakerClient
+
+
+async def main():
+    async with AsyncSageMakerClient() as sage_maker:
+        # Default: 3 attempts for every operation
+        response = await sage_maker.add_association()
+
+        # Override per operation
+        response = await sage_maker.add_association(config_overrides={"retry_max_attempts": 5})
+
+        # Disable retries for this call
+        response = await sage_maker.add_association(config_overrides={"retry_max_attempts": 1})
+```

capo_sagemaker-0.1.0/README.md

@@ -0,0 +1,75 @@
+# Getting Started
+
+## Installation
+
+```
+pip install capo-sagemaker
+```
+
+## Usage
+
+```python
+from capo_sagemaker import AsyncSageMakerClient
+
+
+async def main():
+    async with AsyncSageMakerClient() as sage_maker:
+        # Example: call the add_association operation
+        response = await sage_maker.add_association()
+        print(response["source_arn"])
+```
+
+## Pagination
+
+Some operations in this SDK support pagination. If the operation supports pagination it will have an `iter_` prefixed method that returns an async iterator.
+
+```python
+from capo_sagemaker import AsyncSageMakerClient
+
+
+async def main():
+    async with AsyncSageMakerClient() as sage_maker:
+        # Example: paginate over create_hub_content_presigned_urls
+        async for item in sage_maker.iter_create_hub_content_presigned_urls():
+            print(item)
+```
+
+## Error Handling
+
+The SDK raises exceptions for errors returned by the API. Catch them to handle failures gracefully.
+
+```python
+from capo_sagemaker import AsyncSageMakerClient
+from capo_sagemaker.error import ResourceLimitExceeded
+
+
+async def main():
+    async with AsyncSageMakerClient() as sage_maker:
+        try:
+            await sage_maker.add_association()
+        except ResourceLimitExceeded as e:
+            print(f"Error: {e}")
+            print(e.data)  # additional error data
+```
+
+## Retrying
+
+The SDK retries failed operations automatically. Retry behaviour follows the Smithy specification: errors are retried based on their `is_retryable` and `is_throttling_error` attributes. Throttling errors use a longer base delay. Network-level failures (connection errors and timeouts) are also retried. Non-retryable errors, such as client errors without the `@retryable` trait, are raised immediately without further attempts.
+
+The number of attempts defaults to 3 and can be changed at the client level via `retry_max_attempts`, or per call via `config_overrides`.
+
+```python
+from capo_sagemaker import AsyncSageMakerClient
+
+
+async def main():
+    async with AsyncSageMakerClient() as sage_maker:
+        # Default: 3 attempts for every operation
+        response = await sage_maker.add_association()
+
+        # Override per operation
+        response = await sage_maker.add_association(config_overrides={"retry_max_attempts": 5})
+
+        # Disable retries for this call
+        response = await sage_maker.add_association(config_overrides={"retry_max_attempts": 1})
+```

capo_sagemaker-0.1.0/pyproject.toml

@@ -0,0 +1,61 @@
+[project]
+name = "capo-sagemaker"
+version = "0.1.0"
+description = "Python SDK for Sagemaker."
+readme = "README.md"
+requires-python = ">=3.10"
+classifiers = [
+    "Development Status :: 4 - Beta",
+    "Intended Audience :: Developers",
+    "License :: OSI Approved :: MIT License",
+    "Operating System :: OS Independent",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
+    "Programming Language :: Python :: 3.14",
+    "Topic :: Software Development :: Code Generators",
+    "Typing :: Typed",
+]
+dependencies = [
+    "jmespath>=1.0",
+    "zapros>=0.13.0",
+]
+
+[project.optional-dependencies]
+# IAM Identity Center (SSO), assume-role and web-identity credentials providers
+sso = [
+    "capo-sso>=0.2.0",
+    "capo-sso-oidc>=0.2.0",
+    "capo-sts>=0.3.0",
+]
+
+[tool.pytest.ini_options]
+pythonpath = ["src"]
+testpaths = ["tests"]
+
+[tool.ty.environment]
+python-version = "3.14"
+root = ["./src"]
+
+[tool.ty.src]
+include = ["src", "tests"]
+
+[tool.ty.rules]
+missing-typed-dict-key = "ignore"
+unused-type-ignore-comment = "ignore"
+
+[tool.ruff.lint]
+select = ["E", "F", "I"]
+ignore = ["E501", "E741"]
+
+[dependency-groups]
+dev = [
+    "inline-snapshot>=0.33.0",
+    # TODO: fix for 0.0.69
+    "ty==0.0.68",
+    "pytest>=9.0.3",
+    "ruff>=0.15.14",
+    "trio",
+]

capo_sagemaker-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

capo_sagemaker-0.1.0/src/capo_sagemaker/__init__.py

@@ -0,0 +1,70 @@
+from __future__ import annotations
+
+from . import _iter as _iter
+from . import _protocol as _protocol
+from ._auth._identity import Credentials as Credentials
+from ._auth._identity import Identity as Identity
+from ._auth._providers import (
+    AssumeRoleCredentialsProvider as AssumeRoleCredentialsProvider,
+)
+from ._auth._providers import (
+    AssumeRoleError as AssumeRoleError,
+)
+from ._auth._providers import (
+    CachedProvider as CachedProvider,
+)
+from ._auth._providers import (
+    ChainedProvider as ChainedProvider,
+)
+from ._auth._providers import (
+    CredentialsProvider as CredentialsProvider,
+)
+from ._auth._providers import (
+    EnvCredentialsProvider as EnvCredentialsProvider,
+)
+from ._auth._providers import (
+    IdentityNotFound as IdentityNotFound,
+)
+from ._auth._providers import (
+    IdentityProvider as IdentityProvider,
+)
+from ._auth._providers import (
+    MissingDependencyError as MissingDependencyError,
+)
+from ._auth._providers import (
+    ProfileCredentialsProvider as ProfileCredentialsProvider,
+)
+from ._auth._providers import (
+    SsoCredentialsProvider as SsoCredentialsProvider,
+)
+from ._auth._providers import (
+    SSOError as SSOError,
+)
+from ._auth._providers import (
+    StaticAwsCredentialsProvider as StaticAwsCredentialsProvider,
+)
+from ._auth._providers import (
+    WebIdentityCredentialsProvider as WebIdentityCredentialsProvider,
+)
+from ._auth._signers import Signer as Signer
+from ._auth._signers import SigV4Signer as SigV4Signer
+from ._services._pipeline import (
+    AsyncOperationOptions as AsyncOperationOptions,
+)
+from ._services._pipeline import (
+    AsyncOperationRequest as AsyncOperationRequest,
+)
+from ._services._pipeline import (
+    AsyncOperationResponse as AsyncOperationResponse,
+)
+from ._services._pipeline import (
+    OperationOptions as OperationOptions,
+)
+from ._services._pipeline import (
+    OperationRequest as OperationRequest,
+)
+from ._services._pipeline import (
+    OperationResponse as OperationResponse,
+)
+from ._services.async_sage_maker import AsyncSageMakerClient as AsyncSageMakerClient
+from ._services.sage_maker import SageMakerClient as SageMakerClient

capo_sagemaker-0.1.0/src/capo_sagemaker/_async.py

@@ -0,0 +1,25 @@
+from __future__ import annotations
+
+import asyncio
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    import trio
+else:
+    try:
+        import trio
+    except ImportError:
+        trio = None
+
+
+def in_trio_run() -> bool:
+    if trio is None:
+        return False
+    return trio.lowlevel.in_trio_run()
+
+
+async def anysleep(delay: float) -> None:
+    if in_trio_run():
+        await trio.sleep(delay)
+    else:
+        await asyncio.sleep(delay)

capo_sagemaker-0.1.0/src/capo_sagemaker/_auth/_identity.py

@@ -0,0 +1,16 @@
+from __future__ import annotations
+
+from datetime import datetime
+from typing import TypedDict
+
+from typing_extensions import NotRequired
+
+
+class Identity(TypedDict):
+    expiration: NotRequired[datetime | None]
+
+
+class Credentials(Identity):
+    access_key: str
+    secret_key: str
+    session_token: NotRequired[str | None]