More model_dump

anvilco · Feb 16, 2024 · 5a067f1 · 5a067f1
1 parent 27054bd
commit 5a067f1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/python_anvil/api.py b/python_anvil/api.py
@@ -158,7 +158,7 @@ def fill_pdf(
         api = RestRequest(client=self.client)
         return api.post(
             f"fill/{template_id}.pdf",
-            data.dict(by_alias=True, exclude_none=True) if data else {},
+            data.model_dump(by_alias=True, exclude_none=True) if data else {},
             **kwargs,
         )
 
@@ -180,7 +180,7 @@ def generate_pdf(self, payload: Union[AnyStr, Dict, GeneratePDFPayload], **kwarg
         # Any data errors would come from here
         api = RestRequest(client=self.client)
         return api.post(
-            "generate-pdf", data=data.dict(by_alias=True, exclude_none=True), **kwargs
+            "generate-pdf", data=data.model_dump(by_alias=True, exclude_none=True), **kwargs
         )
 
     def get_cast(
@@ -396,7 +396,7 @@ def create_etch_packet(
             )
 
         payload = mutation.create_payload()
-        variables = payload.dict(by_alias=True, exclude_none=True)
+        variables = payload.model_dump(by_alias=True, exclude_none=True)
 
         return self.mutate(mutation, variables=variables, upload_files=True, **kwargs)
 
@@ -407,7 +407,7 @@ def generate_etch_signing_url(self, signer_eid: str, client_user_id: str, **kwar
             client_user_id=client_user_id,
         )
         payload = mutation.create_payload()
-        return self.mutate(mutation, variables=payload.dict(by_alias=True), **kwargs)
+        return self.mutate(mutation, variables=payload.model_dump(by_alias=True), **kwargs)
 
     def download_documents(self, document_group_eid: str, **kwargs):
         """Retrieve all completed documents in zip form."""
@@ -440,6 +440,6 @@ def forge_submit(
 
         return self.mutate(
             mutation,
-            variables=mutation.create_payload().dict(by_alias=True, exclude_none=True),
+            variables=mutation.create_payload().model_dump(by_alias=True, exclude_none=True),
             **kwargs,
         )
diff --git a/python_anvil/tests/test_api.py b/python_anvil/tests/test_api.py
@@ -348,7 +348,7 @@ def test_adding_unsupported_fields(m_request_post, m_create_unique, anvil):
             assert m_request_post.call_count == 1
 
             variables = m_request_post.call_args[1]["variable_values"]
-            assert cep.dict(by_alias=True, exclude_none=True) == variables
+            assert cep.model_dump(by_alias=True, exclude_none=True) == variables
             assert "newFeature" in variables
 
     def describe_forge_submit():