get rid of df.head and control the inputs instead

This commit is contained in:
Jun-te Kim 2026-03-06 13:57:33 +00:00
parent 58843a5409
commit 8163a97b97

View file

@ -136,7 +136,7 @@ def handler(body: dict[str, Any], context: Any, local: bool = False) -> None:
# Assumption designing with address2uprn was ran first
csv_data = read_csv_from_s3_dict(bucket, key)
df = pd.DataFrame(csv_data)
df = df.head(5)
# df = df.head(5)
# If lexiscore_column is specified, use it; otherwise process all rows
if lexiscore_column and lexiscore_column in df.columns: