Fix mypy in providers/salesforce (#20325)

apache · Dec 16, 2021 · 92bc804 · 92bc804
1 parent 4b8a120
commit 92bc804
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/airflow/providers/salesforce/operators/salesforce_apex_rest.py b/airflow/providers/salesforce/operators/salesforce_apex_rest.py
@@ -41,7 +41,7 @@ def __init__(
         *,
         endpoint: str,
         method: str = 'GET',
-        payload: dict = None,
+        payload: dict,
         salesforce_conn_id: str = 'salesforce_default',
         **kwargs,
     ) -> None:
@@ -59,8 +59,11 @@ def execute(self, context: dict) -> dict:
         :return: Apex response
         :rtype: dict
         """
+        result: dict = {}
         sf_hook = SalesforceHook(salesforce_conn_id=self.salesforce_conn_id)
         conn = sf_hook.get_conn()
-        result = conn.apexecute(action=self.endpoint, method=self.method, data=self.payload)
+        execution_result = conn.apexecute(action=self.endpoint, method=self.method, data=self.payload)
         if self.do_xcom_push:
-            return result
+            result = execution_result
+
+        return result