fix: iac diff scan typing

Author: sylvain-baud-gg

pygitguardian/client.py

@@ -14,7 +14,6 @@
 from .config import DEFAULT_API_VERSION, DEFAULT_BASE_URI, DEFAULT_TIMEOUT
 from .iac_models import (
     IaCDiffScanResult,
-    IaCDiffScanResultSchema,
     IaCScanParameters,
     IaCScanParametersSchema,
     IaCScanResult,
@@ -533,7 +532,7 @@ def iac_diff_scan(
             result.status_code = 504
         else:
             if is_ok(resp):
-                result = IaCDiffScanResultSchema.from_dict(resp.json())  # type: ignore
+                result = IaCDiffScanResult.from_dict(resp.json())
             else:
                 result = load_detail(resp)
 

pygitguardian/iac_models.py

@@ -73,7 +73,7 @@ class IaCDiffScanEntities(Base):
 
 
 @dataclass
-class IaCDiffScanResult(Base):
+class IaCDiffScanResult(Base, FromDictMixin):
     id: str = ""
     type: str = ""
     iac_engine_version: str = ""
@@ -82,6 +82,7 @@ class IaCDiffScanResult(Base):
     )
 
 
-IaCDiffScanResultSchema = marshmallow_dataclass.class_schema(
-    IaCDiffScanResult, BaseSchema
+IaCDiffScanResultSchema = cast(
+    Type[BaseSchema], marshmallow_dataclass.class_schema(IaCDiffScanResult, BaseSchema)
 )
+IaCDiffScanResult.SCHEMA = IaCDiffScanResultSchema()