Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 2 additions & 3 deletions application_sdk/inputs/parquet.py
Original file line number Diff line number Diff line change
Expand Up @@ -89,11 +89,10 @@ async def get_dataframe(self) -> "pd.DataFrame":
try:
import pandas as pd

path = self.path
if self.input_prefix and self.path:
path = await self.download_files(self.path)
await self.download_files(self.path)
# Use pandas native read_parquet which can handle both single files and directories
return pd.read_parquet(path)
return pd.read_parquet(self.path)
Comment on lines +93 to +95
Copy link

Copilot AI Sep 23, 2025

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

The fix assumes that download_files modifies self.path in-place, but this behavior is not evident from the code shown. If download_files doesn't update self.path to point to the downloaded local file, this will still read from the original remote path. Consider either ensuring download_files updates self.path or capturing its return value if it returns the local path.

Copilot uses AI. Check for mistakes.

Comment on lines 92 to +95
Copy link

Copilot AI Sep 23, 2025

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

The fix assumes that download_files modifies self.path in-place, but this behavior is not evident from the code shown. If download_files doesn't update self.path to point to the downloaded local file, this will still read from the original remote path. Consider either ensuring download_files updates self.path or capturing its return value if it returns the local path.

Copilot uses AI. Check for mistakes.

except Exception as e:
logger.error(f"Error reading data from parquet file(s): {str(e)}")
# Re-raise to match IcebergInput behavior
Expand Down