vidhimudaliar commited on
Commit
392a112
·
verified ·
1 Parent(s): f4db987

Update clean.py

Browse files
Files changed (1) hide show
  1. clean.py +3 -0
clean.py CHANGED
@@ -98,6 +98,9 @@ def main():
98
  columns=["filename", "video_id", "video_path", "label", "start", "end"],
99
  )
100
 
 
 
 
101
  cleaned.to_csv(OUTPUT_CSV, index=False)
102
  print(f"✅ Wrote {len(cleaned)} rows -> {OUTPUT_CSV}")
103
 
 
98
  columns=["filename", "video_id", "video_path", "label", "start", "end"],
99
  )
100
 
101
+ cleaned["start"] = cleaned["start"].astype("Int64")
102
+ cleaned["end"] = cleaned["end"].astype("Int64")
103
+
104
  cleaned.to_csv(OUTPUT_CSV, index=False)
105
  print(f"✅ Wrote {len(cleaned)} rows -> {OUTPUT_CSV}")
106