craft-ai-sdk 0.62.2__tar.gz → 0.62.3__tar.gz

{craft_ai_sdk-0.62.2 → craft_ai_sdk-0.62.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: craft-ai-sdk
-Version: 0.62.2
+Version: 0.62.3
 Summary: Craft AI MLOps platform SDK
 License: Apache-2.0
 Author: Craft AI

{craft_ai_sdk-0.62.2 → craft_ai_sdk-0.62.3}/craft_ai_sdk/__init__.py

@@ -13,4 +13,4 @@ from .io import (  # noqa: F401
 )
 from .sdk import CraftAiSdk  # noqa: F401
 
-__version__ = "0.62.2"
+__version__ = "0.62.3"

{craft_ai_sdk-0.62.2 → craft_ai_sdk-0.62.3}/craft_ai_sdk/core/endpoints.py

@@ -23,7 +23,7 @@ def trigger_endpoint(
     sdk: BaseCraftAiSdk,
     endpoint_name,
     endpoint_token,
-    inputs={},
+    inputs=None,
     wait_for_completion=True,
 ):
     """Trigger an endpoint.
@@ -50,6 +50,8 @@ def trigger_endpoint(
           output names as keys and corresponding values as values. Note that this
           key is only returned if ``wait_for_completion`` is `True`.
     """
+    if inputs is None:
+        inputs = {}
     endpoint_url_path = _get_endpoint_url_path(sdk, endpoint_name)
 
     body = {}

{craft_ai_sdk-0.62.2 → craft_ai_sdk-0.62.3}/craft_ai_sdk/core/pipeline_metrics.py

@@ -23,7 +23,8 @@ def record_metric_value(sdk: BaseCraftAiSdk, name, value):
         if sdk.warn_on_metric_outside_of_step:
             warnings.warn(
                 "You cannot send a metric outside a step code, the metric has not \
-been sent"
+been sent",
+                stacklevel=2,
             )
         return False
     url = f"{sdk.base_environment_api_url}" f"/metrics/single-value/{name}"
@@ -51,7 +52,8 @@ def record_list_metric_values(sdk: BaseCraftAiSdk, name, values):
         if sdk.warn_on_metric_outside_of_step:
             warnings.warn(
                 "You cannot send a metric outside a step code, the metric has not \
-been sent"
+been sent",
+                stacklevel=2,
             )
         return False
 

{craft_ai_sdk-0.62.2 → craft_ai_sdk-0.62.3}/craft_ai_sdk/core/pipelines.py

@@ -190,6 +190,7 @@ def create_pipeline(
             "Providing the step name as a positional argument is deprecated and will "
             "be removed in a future version. Please use the step_name keyword argument instead.",  # noqa: E501
             FutureWarning,
+            stacklevel=2,
         )
         return _create_pipeline_with_step(sdk, pipeline_name, function_path)
 

{craft_ai_sdk-0.62.2 → craft_ai_sdk-0.62.3}/craft_ai_sdk/io.py

@@ -155,6 +155,7 @@ class InputSource:
                 "Providing the step_input_name argument is deprecated and will "
                 "be removed in a future version. Please use the pipeline_input_name keyword argument instead.",  # noqa: E501
                 FutureWarning,
+                stacklevel=2,
             )
         self.pipeline_input_name = pipeline_input_name
         self.endpoint_input_name = endpoint_input_name
@@ -243,6 +244,7 @@ class OutputDestination:
                 "Providing the 'step_output_name' argument is deprecated and will "
                 "be removed in a future version. Please use the 'pipeline_output_name' keyword argument instead.",  # noqa: E501
                 FutureWarning,
+                stacklevel=2,
             )
         self.pipeline_output_name = pipeline_output_name
         self.endpoint_output_name = endpoint_output_name

{craft_ai_sdk-0.62.2 → craft_ai_sdk-0.62.3}/craft_ai_sdk/sdk.py

@@ -1,6 +1,7 @@
 import os
 import sys
 import time
+from typing import Any
 import warnings
 from abc import ABC, abstractmethod
 from datetime import timedelta
@@ -24,23 +25,23 @@ class BaseCraftAiSdk(ABC):
     _version: str
 
     @abstractmethod
-    def _get(self, url, params=None, **kwargs):
+    def _get(self, url: str, params=None, **kwargs) -> Any:
         pass
 
     @abstractmethod
-    def _post(self, url, data=None, params=None, files=None, **kwargs):
+    def _post(self, url: str, data=None, params=None, files=None, **kwargs) -> Any:
         pass
 
     @abstractmethod
-    def _put(self, url, data=None, params=None, files=None, **kwargs):
+    def _put(self, url: str, data=None, params=None, files=None, **kwargs) -> Any:
         pass
 
     @abstractmethod
-    def _patch(self, url, data=None, params=None, files=None, **kwargs):
+    def _patch(self, url: str, data=None, params=None, files=None, **kwargs) -> Any:
         pass
 
     @abstractmethod
-    def _delete(self, url, **kwargs):
+    def _delete(self, url: str, **kwargs) -> Any:
         pass
 
 
@@ -136,7 +137,7 @@ class CraftAiSdk(BaseCraftAiSdk):
         os.environ.get("CRAFT_AI__MULTIPART_PART_SIZE__B", str(38 * 256 * 1024))
     )
     _access_token_margin = timedelta(seconds=30)
-    _version = "0.62.2"  # Would be better to share it somewhere
+    _version = "0.62.3"  # Would be better to share it somewhere
 
     def __init__(
         self,

{craft_ai_sdk-0.62.2 → craft_ai_sdk-0.62.3}/craft_ai_sdk/shared/authentication.py

@@ -1,8 +1,11 @@
 import functools
 from datetime import datetime
+from typing import Callable
 
+from requests import Response
 
-def use_authentication(action_func):
+
+def use_authentication(action_func: Callable[..., Response]) -> Callable[..., Response]:
     @functools.wraps(action_func)
     def wrapper(sdk, *args, headers=None, **kwargs):
         actual_headers = None

{craft_ai_sdk-0.62.2 → craft_ai_sdk-0.62.3}/craft_ai_sdk/shared/request_response_handler.py

@@ -1,3 +1,4 @@
+from typing import Any, Callable
 import xml.etree.ElementTree as ET
 from json import JSONDecodeError
 
@@ -7,7 +8,7 @@ from craft_ai_sdk.exceptions import SdkException
 from craft_ai_sdk.shared.execution_context import get_execution_id
 
 
-def handle_data_store_response(response):
+def handle_data_store_response(response: Response):
     """Return the content of a response received from the datastore
     or parse the send error and raise it.
 
@@ -43,7 +44,7 @@ def handle_data_store_response(response):
         )
 
 
-def _parse_json_response(response):
+def _parse_json_response(response: Response):
     if response.status_code == 204 or response.text == "OK":
         return
     try:
@@ -95,8 +96,8 @@ def handle_http_response(response: Response):
     _raise_craft_ai_error_from_response(response)
 
 
-def handle_http_request(request_func):
-    def wrapper(*args, **kwargs):
+def handle_http_request(request_func: Callable[..., Response]) -> Callable[..., Any]:
+    def wrapper(*args, **kwargs) -> Any:
         get_response = kwargs.pop("get_response", False)
         try:
             response = request_func(*args, **kwargs)

{craft_ai_sdk-0.62.2 → craft_ai_sdk-0.62.3}/craft_ai_sdk/shared/warnings.py

@@ -31,7 +31,7 @@ def _documentation_function_warning_decorator(
     def _get_wrapper(func, warning_message):
         @functools.wraps(func)
         def wrapper(*args, **kwargs):
-            warnings.warn(warning_message, warning_category)
+            warnings.warn(warning_message, warning_category, stacklevel=2)
             return func(*args, **kwargs)
 
         doc_start, doc_separator, doc_end = func.__doc__.partition("\n\n")
@@ -111,11 +111,13 @@ def experimental_parameters(parameters):
                         warnings.warn(
                             parameters[key][value]["message"],
                             FutureWarning,
+                            stacklevel=2,
                         )
                     elif EXPERIMENTAL_VALUES.ALL in parameters[key]:
                         warnings.warn(
                             parameters[key][EXPERIMENTAL_VALUES.ALL]["message"],
                             FutureWarning,
+                            stacklevel=2,
                         )
                     elif EXPERIMENTAL_VALUES.STRING_CONTAINING in parameters[key]:
                         if any(
@@ -129,6 +131,7 @@ def experimental_parameters(parameters):
                                     "message"
                                 ],
                                 FutureWarning,
+                                stacklevel=2,
                             )
             return func(*args, **kwargs)
 

{craft_ai_sdk-0.62.2 → craft_ai_sdk-0.62.3}/craft_ai_sdk/utils/file_utils.py

@@ -37,7 +37,7 @@ def convert_size(size_in_bytes):
     Convert a size in bytes to a human readable string.
     """
     units = ["B", "KB", "MB", "GB", "TB"]
-    for unit in units:
+    for unit in units:  # noqa: B007
         if size_in_bytes < 1024.0:
             break
         size_in_bytes /= 1024.0

{craft_ai_sdk-0.62.2 → craft_ai_sdk-0.62.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "craft-ai-sdk"
-version = "0.62.2"
+version = "0.62.3"
 description = "Craft AI MLOps platform SDK"
 license = "Apache-2.0"
 authors = ["Craft AI <contact@craft.ai>"]
@@ -36,6 +36,7 @@ ipython = "^8.18.1"
 ipykernel = "^6.29.5"
 black = "^25.1.0"
 flake8 = "^7.1.2"
+flake8-bugbear = "^24.12.12"
 pylint = "^3.3.4"
 python-dotenv = "^1.0.1"
 pytest = "^8.3.4"