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

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

python-sdk/tests/sql/operators/test_base_decorator.py

@@ -1,4 +1,5 @@
 from unittest import mock
+
 import pytest
 
 from astro.sql import RawSQLOperator
@@ -13,15 +14,11 @@ def test_base_sql_decorated_operator_template_fields_with_parameters():
     assert "parameters" in BaseSQLDecoratedOperator.template_fields
 
 
-@pytest.mark.parametrize(
-    "exception", [OSError("os error"), TypeError("type error")]
-)
+@pytest.mark.parametrize("exception", [OSError("os error"), TypeError("type error")])
 @mock.patch("astro.sql.operators.base_decorator.inspect.getsource", autospec=True)
 def test_get_source_code_handle_exception(mock_getsource, exception):
     """assert get_source_code not raise exception"""
     mock_getsource.side_effect = exception
-    RawSQLOperator(
-        task_id="test",
-        sql="select * from 1",
-        python_callable=lambda: 1
-    ).get_source_code(py_callable=None)
+    RawSQLOperator(task_id="test", sql="select * from 1", python_callable=lambda: 1).get_source_code(
+        py_callable=None
+    )

python-sdk/tests/sql/operators/test_dataframe.py

@@ -12,8 +12,8 @@
 import astro.sql as aql
 from astro.airflow.datasets import DATASET_SUPPORT
 from astro.files import File
-from astro.table import Table
 from astro.sql.operators.dataframe import DataframeOperator
+from astro.table import Table
 
 from ..operators import utils as test_utils
 
@@ -238,15 +238,11 @@ def count_df(df: pandas.DataFrame):
     mock_serde.deserialize.assert_not_called()
 
 
-@pytest.mark.parametrize(
-    "exception", [OSError("os error"), TypeError("type error")]
-)
+@pytest.mark.parametrize("exception", [OSError("os error"), TypeError("type error")])
 @mock.patch("astro.sql.operators.base_decorator.inspect.getsource", autospec=True)
 def test_get_source_code_handle_exception(mock_getsource, exception):
     """assert get_source_code not raise exception"""
     mock_getsource.side_effect = exception
-    DataframeOperator(
-        task_id="test",
-        sql="select * from 1",
-        python_callable=lambda: 1
-    ).get_source_code(py_callable=None)
+    DataframeOperator(task_id="test", sql="select * from 1", python_callable=lambda: 1).get_source_code(
+        py_callable=None
+    )