Fix Google mlengine MyPy errors (#20569)

Part of #19891
apache · Dec 30, 2021 · dda6887 · dda6887
1 parent d56e7b5
commit dda6887
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/airflow/providers/google/cloud/operators/mlengine.py b/airflow/providers/google/cloud/operators/mlengine.py
@@ -19,7 +19,7 @@
 import logging
 import re
 import warnings
-from typing import TYPE_CHECKING, Dict, List, Optional, Sequence, Union
+from typing import TYPE_CHECKING, Any, Dict, List, Optional, Sequence, Union
 
 from airflow.exceptions import AirflowException
 from airflow.models import BaseOperator, BaseOperatorLink
@@ -243,7 +243,7 @@ def __init__(
 
     def execute(self, context: 'Context'):
         job_id = _normalize_mlengine_job_id(self._job_id)
-        prediction_request = {
+        prediction_request: Dict[str, Any] = {
             'jobId': job_id,
             'predictionInput': {
                 'dataFormat': self._data_format,
@@ -1237,7 +1237,7 @@ def __init__(
 
     def execute(self, context: 'Context'):
         job_id = _normalize_mlengine_job_id(self._job_id)
-        training_request = {
+        training_request: Dict[str, Any] = {
             'jobId': job_id,
             'trainingInput': {
                 'scaleTier': self._scale_tier,