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):