From a6858c0bcb29df20e372f4145cc2ccfa9adbd5f2 Mon Sep 17 00:00:00 2001 From: HH Date: Fri, 15 Sep 2023 08:49:34 +0800 Subject: [PATCH] fix the suggestions Signed-off-by: HH --- flytekit/types/structured/__init__.py | 2 +- flytekit/types/structured/snowflake.py | 5 +---- 2 files changed, 2 insertions(+), 5 deletions(-) diff --git a/flytekit/types/structured/__init__.py b/flytekit/types/structured/__init__.py index 617e4bcafa..0783a99077 100644 --- a/flytekit/types/structured/__init__.py +++ b/flytekit/types/structured/__init__.py @@ -82,5 +82,5 @@ def register_snowflake_handlers(): except ImportError: logger.info( "We won't register snowflake handler for structured dataset because " - "we can't find package snowflakee-connector-python" + "we can't find package snowflake-connector-python" ) diff --git a/flytekit/types/structured/snowflake.py b/flytekit/types/structured/snowflake.py index 4842a1c28e..5af2a93430 100644 --- a/flytekit/types/structured/snowflake.py +++ b/flytekit/types/structured/snowflake.py @@ -67,10 +67,7 @@ def _read_from_sf( cs = conn.cursor() cs.execute(f"select * from {table}") - dff = cs.fetch_pandas_all() - print("cs", cs) - print("dff", dff) - return dff + return cs.fetch_pandas_all() class PandasToSnowflakeEncodingHandlers(StructuredDatasetEncoder):