hiddenlayer-sdk 1.1.1__py3-none-any.whl → 1.2.1__py3-none-any.whl

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,
@@ -382,7 +383,7 @@ class ModelScanAPI:
         *,
         model_name: str,
         model_version: Optional[int] = None,
-    ) -> Optional[Sarif]:
+    ) -> Optional[str]:
         """
         Get sarif results from a model scan.
 
@@ -405,8 +406,9 @@ class ModelScanAPI:
         request[2]["Accept"] = "application/sarif+json"
         response = self._api_client.call_api(*request)
         response.read()
+
         return self._api_client.response_deserialize(
-            response_data=response, response_types_map={"200": Sarif}
+            response_data=response, response_types_map={"200": str}
         ).data  # type: ignore
 
     def scan_folder(

hiddenlayer/sdk/version.py

@@ -1 +1 @@
-VERSION = "1.1.1"
+VERSION = "1.2.1"

{hiddenlayer_sdk-1.1.1.dist-info → hiddenlayer_sdk-1.2.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: hiddenlayer-sdk
-Version: 1.1.1
+Version: 1.2.1
 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.1.dist-info → hiddenlayer_sdk-1.2.1.dist-info}/RECORD

@@ -4,7 +4,7 @@ hiddenlayer/sdk/constants.py,sha256=QwBZAQdF7FW9q4C-O8LyxrpqfJFBBlqRUDMjnJ-ouZQ,
 hiddenlayer/sdk/exceptions.py,sha256=LNkD6AvSwNwQjnTDrvY152DmvPKzSejrBsIu-qI44Ec,225
 hiddenlayer/sdk/models.py,sha256=T6GJTFCfBQgBteR0pl0eyYzan-iIFOSpcnlmlCuuQwc,1836
 hiddenlayer/sdk/utils.py,sha256=Ntao8hfsYuB7obZOanayIlrgJ98IWcPhs3sRi39IDkg,2997
-hiddenlayer/sdk/version.py,sha256=mDldA753XXTzV03_TN3RwueHcj_1Dg00eIWo8LRZG5Y,18
+hiddenlayer/sdk/version.py,sha256=QEOrN0FLYbf9G_JHQ21m2esKHMbgVGkm_l_9cpxloj4,18
 hiddenlayer/sdk/rest/__init__.py,sha256=Egcd1PM10KlYQww3y-HkVz0IqF3jnycvn5lYcE6uXVw,7809
 hiddenlayer/sdk/rest/api_client.py,sha256=Zd_yX0nQq06z0Ome2KcsNUcETnOxzhQK9js53qsN7aI,26333
 hiddenlayer/sdk/rest/api_response.py,sha256=eMxw1mpmJcoGZ3gs9z6jM4oYoZ10Gjk333s9sKxGv7s,652
@@ -110,9 +110,9 @@ hiddenlayer/sdk/rest/models/web_response.py,sha256=iwGQ5WjqAB4qGwe9nzGOwtxB1OSUc
 hiddenlayer/sdk/services/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 hiddenlayer/sdk/services/aidr_predictive.py,sha256=aD2anFzWQzfLsR13NwwdQjt_X_iAlOkmss_yHy-154U,2958
 hiddenlayer/sdk/services/model.py,sha256=qoVrwgL7E0raCUI0bNyim4SrmDEGBmcU1UaoJ-JNoCg,4871
-hiddenlayer/sdk/services/model_scan.py,sha256=xGHveChHA6pHArSNF8YKXFOi0FqaH9BnzfuKC_6VyBY,17123
-hiddenlayer_sdk-1.1.1.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-hiddenlayer_sdk-1.1.1.dist-info/METADATA,sha256=quGYg7HvsPUKSD3yu9tTxfTzS692TkN0MP253DrighI,17342
-hiddenlayer_sdk-1.1.1.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-hiddenlayer_sdk-1.1.1.dist-info/top_level.txt,sha256=8BxcmGvEN1RqsMvTWg9Q0vDmtBGcTmatnme6nzyPj2U,12
-hiddenlayer_sdk-1.1.1.dist-info/RECORD,,
+hiddenlayer/sdk/services/model_scan.py,sha256=aMuyAm5atndUyPG46tsxSwsYok4KNIVcTOjxZFFowKI,17176
+hiddenlayer_sdk-1.2.1.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+hiddenlayer_sdk-1.2.1.dist-info/METADATA,sha256=AK__e40yjqBWBq_Y4SEExim80vYlG68cAFUz4CWgEBM,17342
+hiddenlayer_sdk-1.2.1.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+hiddenlayer_sdk-1.2.1.dist-info/top_level.txt,sha256=8BxcmGvEN1RqsMvTWg9Q0vDmtBGcTmatnme6nzyPj2U,12
+hiddenlayer_sdk-1.2.1.dist-info/RECORD,,