Skip to content

Commit 5462138

Browse files
sdks/python: fix linting issues
1 parent 84d5e4f commit 5462138

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

sdks/python/apache_beam/transforms/enrichment_handlers/cloudsql.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -316,7 +316,11 @@ def __enter__(self):
316316
self._engine = create_engine(
317317
url=self._connection_config.get_db_url(), creator=connector)
318318

319-
def _execute_query(self, query: str, is_batch: bool, **params):
319+
def _execute_query(
320+
self,
321+
query: str,
322+
is_batch: bool,
323+
**params) -> Union[List[Dict[str, Any]], Dict[str, Any]]:
320324
connection = None
321325
try:
322326
connection = self._engine.connect()

sdks/python/apache_beam/transforms/enrichment_handlers/cloudsql_it_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -90,7 +90,7 @@ def start_sql_db_container(
9090
sql_client_retries=3) -> Optional[SQLDBContainerInfo]:
9191
info = None
9292
for i in range(sql_client_retries):
93-
sql_db_container = None
93+
sql_db_container = DbContainer("")
9494
try:
9595
if database_type == DatabaseTypeAdapter.POSTGRESQL:
9696
user, password, db_id = "test", "test", "test"

0 commit comments

Comments
 (0)