Skip to content

Commit

Permalink
fix the suggestions
Browse files Browse the repository at this point in the history
Signed-off-by: HH <hhcs9527@gmail.com>
  • Loading branch information
hhcs9527 committed Sep 27, 2023
1 parent 62f9170 commit a6858c0
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 5 deletions.
2 changes: 1 addition & 1 deletion flytekit/types/structured/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -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"
)
5 changes: 1 addition & 4 deletions flytekit/types/structured/snowflake.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down

0 comments on commit a6858c0

Please sign in to comment.