@@ -44,7 +44,7 @@ def file_transformation(
4444 self ._audit_tables .mark_failed (submissions = [submission_info .submission_id ])
4545 return submission_info , SubmissionStatus (processing_failed = True )
4646
47- def apply_data_contract (self , submission_info : SubmissionInfo , submission_status : SubmissionStatus ) -> tuple [SubmissionInfo | SubmissionStatus ]:
47+ def apply_data_contract (self , submission_info : SubmissionInfo , submission_status : Optional [ SubmissionStatus ] = None ) -> tuple [SubmissionInfo | SubmissionStatus ]:
4848 try :
4949 return super ().apply_data_contract (submission_info , submission_status )
5050 except Exception as exc : # pylint: disable=W0718
@@ -58,7 +58,7 @@ def apply_data_contract(self, submission_info: SubmissionInfo, submission_status
5858 self ._audit_tables .mark_failed (submissions = [submission_info .submission_id ])
5959 return submission_info , SubmissionStatus (processing_failed = True )
6060
61- def apply_business_rules (self , submission_info : SubmissionInfo , submission_status : SubmissionStatus ):
61+ def apply_business_rules (self , submission_info : SubmissionInfo , submission_status : Optional [ SubmissionStatus ] = None ):
6262 try :
6363 return super ().apply_business_rules (submission_info , submission_status )
6464 except Exception as exc : # pylint: disable=W0718
@@ -72,7 +72,7 @@ def apply_business_rules(self, submission_info: SubmissionInfo, submission_statu
7272 self ._audit_tables .mark_failed (submissions = [submission_info .submission_id ])
7373 return submission_info , SubmissionStatus (processing_failed = True )
7474
75- def error_report (self , submission_info : SubmissionInfo , submission_status : SubmissionStatus ):
75+ def error_report (self , submission_info : SubmissionInfo , submission_status : Optional [ SubmissionStatus ] = None ):
7676 try :
7777 return super ().error_report (submission_info , submission_status )
7878 except Exception as exc : # pylint: disable=W0718
0 commit comments