🚨 Fix mypy warnings.

python-lapidary · Jan 17, 2024 · e673611 · e673611
1 parent 7a2bb51
commit e673611
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/src/lapidary/runtime/response.py b/src/lapidary/runtime/response.py
@@ -18,7 +18,7 @@
 def parse_model(response: httpx.Response, typ: ty.Type[T]) -> T:
     if inspect.isclass(typ):
         if issubclass(typ, pydantic.BaseModel):
-            return typ.model_validate_json(response.content)
+            return ty.cast(T, typ.model_validate_json(response.content))
 
     return pydantic.TypeAdapter(typ).validate_json(response.content)
 

diff --git a/tests/test_client.py b/tests/test_client.py
@@ -1,3 +1,5 @@
+# mypy: disable-error-code="empty-body"
+
 import dataclasses as dc
 import unittest
 

diff --git a/tests/test_request.py b/tests/test_request.py
@@ -24,7 +24,7 @@ class MyRequestBodyList(pydantic.RootModel):
         request_factory = Mock()
         build_request(
             operation=OperationModel('GET', 'path', {}, RequestBodyModel('body', {'application/json': MyRequestBodyList}), {}),
-            actual_params={'body': MyRequestBodyList([MyRequestBodyModel(a='a')])},
+            actual_params={'body': MyRequestBodyList(root=[MyRequestBodyModel(a='a')])},
             request_factory=request_factory
         )