[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

airflow/api_connexion/endpoints/pool_endpoint.py

@@ -88,9 +88,11 @@ def patch_pool(pool_name, session, update_mask=None):
         _patch_body = {}
         try:
             update_mask = [
-                pool_schema.declared_fields[field].attribute
-                if pool_schema.declared_fields[field].attribute
-                else field
+                (
+                    pool_schema.declared_fields[field].attribute
+                    if pool_schema.declared_fields[field].attribute
+                    else field
+                )
                 for field in update_mask
             ]
         except KeyError as err:

airflow/providers/amazon/aws/example_dags/example_s3_to_redshift.py

@@ -75,7 +75,11 @@ def _remove_sample_data_from_s3():
     teardown__task_remove_sample_data_from_s3 = PythonOperator(
         python_callable=_remove_sample_data_from_s3, task_id='teardown__remove_sample_data_from_s3'
     )
-    [setup__task_add_sample_data_to_s3, setup__task_create_table] >> task_transfer_s3_to_redshift >> [
-        teardown__task_drop_table,
-        teardown__task_remove_sample_data_from_s3,
-    ]
+    (
+        [setup__task_add_sample_data_to_s3, setup__task_create_table]
+        >> task_transfer_s3_to_redshift
+        >> [
+            teardown__task_drop_table,
+            teardown__task_remove_sample_data_from_s3,
+        ]
+    )

airflow/providers/dingding/example_dags/example_dingding.py

@@ -205,12 +205,19 @@ def failure_callback(context):
         message="",
     )
 
-    [
-        text_msg_remind_none,
-        text_msg_remind_specific,
-        text_msg_remind_include_invalid,
-        text_msg_remind_all,
-    ] >> link_msg >> markdown_msg >> [
-        single_action_card_msg,
-        multi_action_card_msg,
-    ] >> feed_card_msg >> msg_failure_callback
+    (
+        [
+            text_msg_remind_none,
+            text_msg_remind_specific,
+            text_msg_remind_include_invalid,
+            text_msg_remind_all,
+        ]
+        >> link_msg
+        >> markdown_msg
+        >> [
+            single_action_card_msg,
+            multi_action_card_msg,
+        ]
+        >> feed_card_msg
+        >> msg_failure_callback
+    )

airflow/providers/elasticsearch/log/es_task_handler.py

@@ -253,7 +253,7 @@ def set_context(self, ti: TaskInstance) -> None:
                     'execution_date': self._clean_execution_date(ti.execution_date),
                     'try_number': str(ti.try_number),
                     'log_id': self._render_log_id(ti, ti.try_number),
-                    'offset': int(time() * (10 ** 9)),
+                    'offset': int(time() * (10**9)),
                 },
             )
 

airflow/providers/google/cloud/example_dags/example_bigquery_operations.py

@@ -168,10 +168,18 @@
 
     create_dataset >> patch_dataset >> update_dataset >> get_dataset >> get_dataset_result >> delete_dataset
 
-    update_dataset >> create_table >> create_view >> [
-        get_dataset_tables,
-        delete_view,
-    ] >> update_table >> delete_table >> delete_dataset
+    (
+        update_dataset
+        >> create_table
+        >> create_view
+        >> [
+            get_dataset_tables,
+            delete_view,
+        ]
+        >> update_table
+        >> delete_table
+        >> delete_dataset
+    )
     update_dataset >> create_external_table >> delete_dataset
 
 with models.DAG(

airflow/providers/google/cloud/hooks/bigquery.py

@@ -2755,7 +2755,7 @@ def _bind_parameters(operation: str, parameters: dict) -> str:
     """Helper method that binds parameters to a SQL query"""
     # inspired by MySQL Python Connector (conversion.py)
     string_parameters = {}  # type Dict[str, str]
-    for (name, value) in parameters.items():
+    for name, value in parameters.items():
         if value is None:
             string_parameters[name] = 'NULL'
         elif isinstance(value, str):

airflow/providers/google/cloud/hooks/mlengine.py

@@ -62,13 +62,13 @@ def _poll_with_exponential_delay(request, execute_num_retries, max_n, is_done_fu
                 log.info('Operation is done: %s', response)
                 return response
 
-            time.sleep((2 ** i) + (random.randint(0, 1000) / 1000))
+            time.sleep((2**i) + (random.randint(0, 1000) / 1000))
         except HttpError as e:
             if e.resp.status != 429:
                 log.info('Something went wrong. Not retrying: %s', format(e))
                 raise
             else:
-                time.sleep((2 ** i) + (random.randint(0, 1000) / 1000))
+                time.sleep((2**i) + (random.randint(0, 1000) / 1000))
 
     raise ValueError('Connection could not be established after {} retries.'.format(max_n))
 

airflow/providers/mongo/hooks/mongo.py

@@ -55,9 +55,11 @@ def __init__(self, conn_id: str = 'mongo_default', *args, **kwargs) -> None:
 
         self.uri = '{scheme}://{creds}{host}{port}/{database}'.format(
             scheme=scheme,
-            creds='{}:{}@'.format(self.connection.login, self.connection.password)
-            if self.connection.login
-            else '',
+            creds=(
+                '{}:{}@'.format(self.connection.login, self.connection.password)
+                if self.connection.login
+                else ''
+            ),
             host=self.connection.host,
             port='' if self.connection.port is None else ':{}'.format(self.connection.port),
             database=self.connection.schema,

airflow/providers/snowflake/example_dags/example_snowflake.py

@@ -123,9 +123,13 @@
 
 # [END howto_operator_snowflake_to_slack]
 
-snowflake_op_sql_str >> [
-    snowflake_op_with_params,
-    snowflake_op_sql_list,
-    snowflake_op_template_file,
-    copy_into_table,
-] >> slack_report
+(
+    snowflake_op_sql_str
+    >> [
+        snowflake_op_with_params,
+        snowflake_op_sql_list,
+        snowflake_op_template_file,
+        copy_into_table,
+    ]
+    >> slack_report
+)

tests/providers/apache/hive/transfers/test_s3_to_hive.py

@@ -212,7 +212,7 @@ def test_execute(self, mock_hiveclihook):
         conn.create_bucket(Bucket='bucket')
 
         # Testing txt, zip, bz2 files with and without header row
-        for (ext, has_header) in product(['.txt', '.gz', '.bz2', '.GZ'], [True, False]):
+        for ext, has_header in product(['.txt', '.gz', '.bz2', '.GZ'], [True, False]):
             self.kwargs['headers'] = has_header
             self.kwargs['check_headers'] = has_header
             logging.info("Testing %s format %s header", ext, 'with' if has_header else 'without')
@@ -248,7 +248,7 @@ def test_execute_with_select_expression(self, mock_hiveclihook):
         # Only testing S3ToHiveTransfer calls S3Hook.select_key with
         # the right parameters and its execute method succeeds here,
         # since Moto doesn't support select_object_content as of 1.3.2.
-        for (ext, has_header) in product(['.txt', '.gz', '.GZ'], [True, False]):
+        for ext, has_header in product(['.txt', '.gz', '.GZ'], [True, False]):
             input_compressed = ext.lower() != '.txt'
             key = self.s3_key + ext
 

tests/providers/jenkins/hooks/test_jenkins.py

@@ -25,7 +25,7 @@
 class TestJenkinsHook(unittest.TestCase):
     @mock.patch('airflow.hooks.base_hook.BaseHook.get_connection')
     def test_client_created_default_http(self, get_connection_mock):
-        """tests `init` method to validate http client creation when all parameters are passed """
+        """tests `init` method to validate http client creation when all parameters are passed"""
         default_connection_id = 'jenkins_default'
 
         connection_host = 'http://test.com'

tests/providers/slack/hooks/test_slack.py

@@ -27,7 +27,7 @@
 
 class TestSlackHook(unittest.TestCase):
     def test_get_token_with_token_only(self):
-        """tests `__get_token` method when only token is provided """
+        """tests `__get_token` method when only token is provided"""
         # Given
         test_token = 'test_token'
         test_conn_id = None
@@ -43,7 +43,7 @@ def test_get_token_with_token_only(self):
     @mock.patch('airflow.providers.slack.hooks.slack.WebClient')
     @mock.patch('airflow.providers.slack.hooks.slack.SlackHook.get_connection')
     def test_get_token_with_valid_slack_conn_id_only(self, get_connection_mock, mock_slack_client):
-        """tests `__get_token` method when only connection is provided """
+        """tests `__get_token` method when only connection is provided"""
         # Given
         test_token = None
         test_conn_id = 'x'
@@ -63,7 +63,7 @@ def test_get_token_with_valid_slack_conn_id_only(self, get_connection_mock, mock
 
     @mock.patch('airflow.providers.slack.hooks.slack.SlackHook.get_connection')
     def test_get_token_with_no_password_slack_conn_id_only(self, get_connection_mock):
-        """tests `__get_token` method when only connection is provided """
+        """tests `__get_token` method when only connection is provided"""
 
         # Mock
         conn = mock.Mock()
@@ -75,7 +75,7 @@ def test_get_token_with_no_password_slack_conn_id_only(self, get_connection_mock
 
     @mock.patch('airflow.providers.slack.hooks.slack.SlackHook.get_connection')
     def test_get_token_with_empty_password_slack_conn_id_only(self, get_connection_mock):
-        """tests `__get_token` method when only connection is provided """
+        """tests `__get_token` method when only connection is provided"""
 
         # Mock
         get_connection_mock.return_value = mock.Mock(password=None)
@@ -84,7 +84,7 @@ def test_get_token_with_empty_password_slack_conn_id_only(self, get_connection_m
         self.assertRaises(AirflowException, SlackHook, token=None, slack_conn_id='x')
 
     def test_get_token_with_token_and_slack_conn_id(self):
-        """tests `__get_token` method when both arguments are provided """
+        """tests `__get_token` method when both arguments are provided"""
         # Given
         test_token = 'test_token'
         test_conn_id = 'x'
@@ -98,7 +98,7 @@ def test_get_token_with_token_and_slack_conn_id(self):
         self.assertEqual(output, expected)
 
     def test_get_token_with_out_token_nor_slack_conn_id(self):
-        """tests `__get_token` method when no arguments are provided """
+        """tests `__get_token` method when no arguments are provided"""
 
         self.assertRaises(AirflowException, SlackHook, token=None, slack_conn_id=None)
 

tests/providers/yandex/hooks/test_yandex.py

@@ -27,7 +27,7 @@ class TestYandexHook(unittest.TestCase):
     @mock.patch('airflow.hooks.base_hook.BaseHook.get_connection')
     @mock.patch('airflow.providers.yandex.hooks.yandex.YandexCloudBaseHook._get_credentials')
     def test_client_created_without_exceptions(self, get_credentials_mock, get_connection_mock):
-        """tests `init` method to validate client creation when all parameters are passed """
+        """tests `init` method to validate client creation when all parameters are passed"""
 
         # Inputs to constructor
         default_folder_id = 'test_id'
@@ -46,7 +46,6 @@ def test_client_created_without_exceptions(self, get_credentials_mock, get_conne
 
     @mock.patch('airflow.hooks.base_hook.BaseHook.get_connection')
     def test_get_credentials_raise_exception(self, get_connection_mock):
-
         """tests 'get_credentials' method raising exception if none of the required fields are passed."""
 
         # Inputs to constructor