hiddenlayer-sdk 1.1.0__tar.gz → 1.2.0__tar.gz

{hiddenlayer_sdk-1.1.0 → hiddenlayer_sdk-1.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: hiddenlayer-sdk
-Version: 1.1.0
+Version: 1.2.0
 Summary: Official HiddenLayer Python SDK
 Author-email: HiddenLayer Integrations Team <integrations@hiddenlayer.com>
 Maintainer-email: HiddenLayer Integrations Team <integrations@hiddenlayer.com>

{hiddenlayer_sdk-1.1.0 → hiddenlayer_sdk-1.2.0}/hiddenlayer/sdk/rest/api/sensor_api.py

@@ -16,8 +16,8 @@ from pydantic import validate_call, Field, StrictFloat, StrictStr, StrictInt
 from typing import Any, Dict, List, Optional, Tuple, Union
 from typing_extensions import Annotated
 
-from pydantic import Field, StrictFloat, StrictInt, StrictStr, field_validator
-from typing import Any, Optional, Union
+from pydantic import Field, StrictInt, StrictStr, field_validator
+from typing import Any, Optional
 from typing_extensions import Annotated
 from hiddenlayer.sdk.rest.models.create_sensor_request import CreateSensorRequest
 from hiddenlayer.sdk.rest.models.get_multipart_upload_response import GetMultipartUploadResponse
@@ -48,7 +48,7 @@ class SensorApi:
     def begin_multipart_upload(
         self,
         sensor_id: StrictStr,
-        x_content_length: Annotated[Union[StrictFloat, StrictInt], Field(description="The total size of multipart upload.")],
+        x_content_length: Annotated[StrictInt, Field(description="The total size of multipart upload.")],
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -68,7 +68,7 @@ class SensorApi:
         :param sensor_id: (required)
         :type sensor_id: str
         :param x_content_length: The total size of multipart upload. (required)
-        :type x_content_length: float
+        :type x_content_length: int
         :param _request_timeout: timeout setting for this request. If one
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
@@ -119,7 +119,7 @@ class SensorApi:
     def begin_multipart_upload_with_http_info(
         self,
         sensor_id: StrictStr,
-        x_content_length: Annotated[Union[StrictFloat, StrictInt], Field(description="The total size of multipart upload.")],
+        x_content_length: Annotated[StrictInt, Field(description="The total size of multipart upload.")],
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -139,7 +139,7 @@ class SensorApi:
         :param sensor_id: (required)
         :type sensor_id: str
         :param x_content_length: The total size of multipart upload. (required)
-        :type x_content_length: float
+        :type x_content_length: int
         :param _request_timeout: timeout setting for this request. If one
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
@@ -190,7 +190,7 @@ class SensorApi:
     def begin_multipart_upload_without_preload_content(
         self,
         sensor_id: StrictStr,
-        x_content_length: Annotated[Union[StrictFloat, StrictInt], Field(description="The total size of multipart upload.")],
+        x_content_length: Annotated[StrictInt, Field(description="The total size of multipart upload.")],
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -210,7 +210,7 @@ class SensorApi:
         :param sensor_id: (required)
         :type sensor_id: str
         :param x_content_length: The total size of multipart upload. (required)
-        :type x_content_length: float
+        :type x_content_length: int
         :param _request_timeout: timeout setting for this request. If one
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
@@ -1978,7 +1978,7 @@ class SensorApi:
         self,
         sensor_id: StrictStr,
         upload_id: StrictStr,
-        part: Union[StrictFloat, StrictInt],
+        part: StrictInt,
         body: Optional[Any],
         _request_timeout: Union[
             None,
@@ -2001,7 +2001,7 @@ class SensorApi:
         :param upload_id: (required)
         :type upload_id: str
         :param part: (required)
-        :type part: float
+        :type part: int
         :param body: (required)
         :type body: object
         :param _request_timeout: timeout setting for this request. If one
@@ -2057,7 +2057,7 @@ class SensorApi:
         self,
         sensor_id: StrictStr,
         upload_id: StrictStr,
-        part: Union[StrictFloat, StrictInt],
+        part: StrictInt,
         body: Optional[Any],
         _request_timeout: Union[
             None,
@@ -2080,7 +2080,7 @@ class SensorApi:
         :param upload_id: (required)
         :type upload_id: str
         :param part: (required)
-        :type part: float
+        :type part: int
         :param body: (required)
         :type body: object
         :param _request_timeout: timeout setting for this request. If one
@@ -2136,7 +2136,7 @@ class SensorApi:
         self,
         sensor_id: StrictStr,
         upload_id: StrictStr,
-        part: Union[StrictFloat, StrictInt],
+        part: StrictInt,
         body: Optional[Any],
         _request_timeout: Union[
             None,
@@ -2159,7 +2159,7 @@ class SensorApi:
         :param upload_id: (required)
         :type upload_id: str
         :param part: (required)
-        :type part: float
+        :type part: int
         :param body: (required)
         :type body: object
         :param _request_timeout: timeout setting for this request. If one

{hiddenlayer_sdk-1.1.0 → hiddenlayer_sdk-1.2.0}/hiddenlayer/sdk/rest/models/model_scan_api_v3_scan_query200_response.py

@@ -18,7 +18,7 @@ import re  # noqa: F401
 import json
 
 from pydantic import BaseModel, ConfigDict, Field
-from typing import Any, ClassVar, Dict, List, Optional, Union
+from typing import Any, ClassVar, Dict, List, Optional
 from typing_extensions import Annotated
 from hiddenlayer.sdk.rest.models.scan_report_v3 import ScanReportV3
 from typing import Optional, Set
@@ -29,7 +29,7 @@ class ModelScanApiV3ScanQuery200Response(BaseModel):
     ModelScanApiV3ScanQuery200Response
     """ # noqa: E501
     items: Optional[List[ScanReportV3]] = None
-    total: Union[Annotated[float, Field(strict=True, ge=0)], Annotated[int, Field(strict=True, ge=0)]] = Field(description="Total number of items available based on the query criteria.")
+    total: Annotated[int, Field(strict=True, ge=0)] = Field(description="Total number of items available based on the query criteria.")
     limit: Annotated[int, Field(le=100, strict=True, ge=1)] = Field(description="Maximum number of items to return")
     offset: Annotated[int, Field(strict=True, ge=0)] = Field(description="Begin returning the results from this offset")
     __properties: ClassVar[List[str]] = ["items", "total", "limit", "offset"]

{hiddenlayer_sdk-1.1.0 → hiddenlayer_sdk-1.2.0}/hiddenlayer/sdk/rest/models/multipart_upload_part.py

@@ -17,8 +17,8 @@ import pprint
 import re  # noqa: F401
 import json
 
-from pydantic import BaseModel, ConfigDict, Field, StrictFloat, StrictInt, StrictStr
-from typing import Any, ClassVar, Dict, List, Optional, Union
+from pydantic import BaseModel, ConfigDict, Field, StrictInt, StrictStr
+from typing import Any, ClassVar, Dict, List, Optional
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -26,9 +26,9 @@ class MultipartUploadPart(BaseModel):
     """
     MultipartUploadPart
     """ # noqa: E501
-    part_number: Union[StrictFloat, StrictInt]
-    start_offset: Union[StrictFloat, StrictInt]
-    end_offset: Union[StrictFloat, StrictInt]
+    part_number: StrictInt
+    start_offset: StrictInt
+    end_offset: StrictInt
     upload_url: Optional[StrictStr] = Field(default=None, description="only provided when part is to be directly uploaded to a cloud provider (adhoc)")
     __properties: ClassVar[List[str]] = ["part_number", "start_offset", "end_offset", "upload_url"]
 

{hiddenlayer_sdk-1.1.0 → hiddenlayer_sdk-1.2.0}/hiddenlayer/sdk/rest/models/paged_response_with_total.py

@@ -18,7 +18,7 @@ import re  # noqa: F401
 import json
 
 from pydantic import BaseModel, ConfigDict, Field, StrictStr
-from typing import Any, ClassVar, Dict, List, Optional, Union
+from typing import Any, ClassVar, Dict, List, Optional
 from typing_extensions import Annotated
 from typing import Optional, Set
 from typing_extensions import Self
@@ -28,7 +28,7 @@ class PagedResponseWithTotal(BaseModel):
     PagedResponseWithTotal
     """ # noqa: E501
     items: Optional[List[StrictStr]] = Field(default=None, description="List of items. If no matching items are found, then `[]` will be returned.")
-    total: Union[Annotated[float, Field(strict=True, ge=0)], Annotated[int, Field(strict=True, ge=0)]] = Field(description="Total number of items available based on the query criteria.")
+    total: Annotated[int, Field(strict=True, ge=0)] = Field(description="Total number of items available based on the query criteria.")
     limit: Annotated[int, Field(le=100, strict=True, ge=1)] = Field(description="Maximum number of items to return")
     offset: Annotated[int, Field(strict=True, ge=0)] = Field(description="Begin returning the results from this offset")
     __properties: ClassVar[List[str]] = ["items", "total", "limit", "offset"]

{hiddenlayer_sdk-1.1.0 → hiddenlayer_sdk-1.2.0}/hiddenlayer/sdk/rest/models/scan_detection_v3.py

@@ -66,8 +66,8 @@ class ScanDetectionV3(BaseModel):
         if value is None:
             return value
 
-        if not re.match(r"^CWE-\d{1,4}.*$", value):
-            raise ValueError(r"must validate the regular expression /^CWE-\d{1,4}.*$/")
+        if not re.match(r"^CWE-\d{1,4}.*$|^$", value):
+            raise ValueError(r"must validate the regular expression /^CWE-\d{1,4}.*$|^$/")
         return value
 
     model_config = ConfigDict(

{hiddenlayer_sdk-1.1.0 → hiddenlayer_sdk-1.2.0}/hiddenlayer/sdk/services/model_scan.py

@@ -261,6 +261,7 @@ class ModelScanAPI:
         self,
         *,
         repo_id: str,
+        model_name: Optional[str] = None,
         # model_id: str,
         # HF parameters
         revision: Optional[str] = None,
@@ -317,7 +318,7 @@ class ModelScanAPI:
         )
 
         return self.scan_folder(
-            model_name=repo_id,
+            model_name=model_name or repo_id,
             path=local_dir,
             allow_file_patterns=allow_file_patterns,
             ignore_file_patterns=ignore_file_patterns,

hiddenlayer_sdk-1.2.0/hiddenlayer/sdk/version.py

@@ -0,0 +1 @@
+VERSION = "1.2.0"

{hiddenlayer_sdk-1.1.0 → hiddenlayer_sdk-1.2.0}/hiddenlayer_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: hiddenlayer-sdk
-Version: 1.1.0
+Version: 1.2.0
 Summary: Official HiddenLayer Python SDK
 Author-email: HiddenLayer Integrations Team <integrations@hiddenlayer.com>
 Maintainer-email: HiddenLayer Integrations Team <integrations@hiddenlayer.com>

hiddenlayer_sdk-1.1.0/hiddenlayer/sdk/version.py

@@ -1 +0,0 @@
-VERSION = "1.1.0"