From 2e20dbb0fe24fbb55d2abaed36e7cc549796b4ed Mon Sep 17 00:00:00 2001
From: shamoon <4887959+shamoon@users.noreply.github.com>
Date: Sat, 30 Dec 2023 00:13:37 -0800
Subject: [PATCH] Fix some type annotations

---
 src/documents/matching.py    | 4 +++-
 src/documents/serialisers.py | 5 ++---
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/src/documents/matching.py b/src/documents/matching.py
index 2592abd6a..28bc783d6 100644
--- a/src/documents/matching.py
+++ b/src/documents/matching.py
@@ -1,8 +1,10 @@
 import logging
 import re
 from fnmatch import fnmatch
+from typing import Union
 
 from documents.classifier import DocumentClassifier
+from documents.data_models import ConsumableDocument
 from documents.data_models import DocumentSource
 from documents.models import Correspondent
 from documents.models import Document
@@ -238,7 +240,7 @@ def _split_match(matching_model):
 
 
 def document_matches_workflow(
-    document,  # ConsumableDocument | Document
+    document: Union[ConsumableDocument, Document],
     workflow: Workflow,
     trigger_type: WorkflowTrigger.WorkflowTriggerType,
 ) -> bool:
diff --git a/src/documents/serialisers.py b/src/documents/serialisers.py
index ea7fd142a..df48a78a8 100644
--- a/src/documents/serialisers.py
+++ b/src/documents/serialisers.py
@@ -3,7 +3,6 @@ import math
 import re
 import zoneinfo
 from decimal import Decimal
-from typing import Any
 
 import magic
 from celery import states
@@ -1451,7 +1450,7 @@ class WorkflowSerializer(serializers.ModelSerializer):
             if action.workflows.all().count() == 0:
                 action.delete()
 
-    def create(self, validated_data: Any) -> Workflow:
+    def create(self, validated_data) -> Workflow:
         if "triggers" in validated_data:
             triggers = validated_data.pop("triggers")
 
@@ -1464,7 +1463,7 @@ class WorkflowSerializer(serializers.ModelSerializer):
 
         return instance
 
-    def update(self, instance: Any, validated_data: Any) -> Workflow:
+    def update(self, instance: Workflow, validated_data) -> Workflow:
         if "triggers" in validated_data:
             triggers = validated_data.pop("triggers")