You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@airflow.apache.org by po...@apache.org on 2021/12/30 16:18:55 UTC
[airflow] branch main updated: Fix Google mlengine MyPy errors (#20569)
This is an automated email from the ASF dual-hosted git repository.
potiuk pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git
The following commit(s) were added to refs/heads/main by this push:
new dda6887 Fix Google mlengine MyPy errors (#20569)
dda6887 is described below
commit dda688776c196eea708abfda0462af6c92809031
Author: Jarek Potiuk <ja...@potiuk.com>
AuthorDate: Thu Dec 30 17:18:25 2021 +0100
Fix Google mlengine MyPy errors (#20569)
Part of #19891
---
airflow/providers/google/cloud/operators/mlengine.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/airflow/providers/google/cloud/operators/mlengine.py b/airflow/providers/google/cloud/operators/mlengine.py
index 777ac10..ecd813d 100644
--- 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 @@ class MLEngineStartBatchPredictionJobOperator(BaseOperator):
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 @@ class MLEngineStartTrainingJobOperator(BaseOperator):
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,