From caf317c5459a5d514741a54dd6c55b1867e3e171 Mon Sep 17 00:00:00 2001
From: jasquat <jasquat@users.noreply.github.com>
Date: Fri, 24 Jun 2022 15:29:22 -0400
Subject: [PATCH] monkeytype w/ burnettk

---
 src/spiffworkflow_backend/models/process_instance.py           | 2 +-
 src/spiffworkflow_backend/services/process_instance_service.py | 2 +-
 tests/spiffworkflow_backend/integration/test_process_api.py    | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/spiffworkflow_backend/models/process_instance.py b/src/spiffworkflow_backend/models/process_instance.py
index 88dc88fb..51687c83 100644
--- a/src/spiffworkflow_backend/models/process_instance.py
+++ b/src/spiffworkflow_backend/models/process_instance.py
@@ -131,7 +131,7 @@ class ProcessInstanceModelSchema(Schema):
 
     status = marshmallow.fields.Method("get_status", dump_only=True)
 
-    def get_status(self, obj):
+    def get_status(self, obj: ProcessInstanceModel) -> str:
         """Get_status."""
         return obj.status
 
diff --git a/src/spiffworkflow_backend/services/process_instance_service.py b/src/spiffworkflow_backend/services/process_instance_service.py
index 1aa40fdb..4897bab3 100644
--- a/src/spiffworkflow_backend/services/process_instance_service.py
+++ b/src/spiffworkflow_backend/services/process_instance_service.py
@@ -126,7 +126,7 @@ class ProcessInstanceService:
         else:
             return {}
 
-    def get_process_instance(self, process_instance_id):
+    def get_process_instance(self, process_instance_id: int) -> ProcessInstanceModel:
         """Get_process_instance."""
         result = (
             db.session.query(ProcessInstanceModel)
diff --git a/tests/spiffworkflow_backend/integration/test_process_api.py b/tests/spiffworkflow_backend/integration/test_process_api.py
index c818ce33..445fc468 100644
--- a/tests/spiffworkflow_backend/integration/test_process_api.py
+++ b/tests/spiffworkflow_backend/integration/test_process_api.py
@@ -576,7 +576,7 @@ def test_process_instance_create(
 
 def test_process_instance_run(
     app: Flask, client: FlaskClient, with_bpmn_file_cleanup: None
-):
+) -> None:
     """Test_process_instance_run."""
     process_group_id = "runs_without_input"
     process_model_id = "sample"