Spaces:

rerun
/

preview_dataset

Running

Emil Ernerfeldt commited on Apr 17, 2024

Commit

d63b041

1 Parent(s): 147aaec

Fix typo

Files changed (1) hide show

main.py CHANGED Viewed

@@ -71,7 +71,7 @@ def main():
     print("Loading dataset…")
     dataset = load_dataset(args.dataset, split="train")
-    print("Selecting episode {args.episode_id}…")
     ds_subset = dataset.filter(lambda frame: frame["episode_id"] == args.episode_id)
     print("Starting Rerun…")

     print("Loading dataset…")
     dataset = load_dataset(args.dataset, split="train")
+    print(f"Selecting episode {args.episode_id}…")
     ds_subset = dataset.filter(lambda frame: frame["episode_id"] == args.episode_id)
     print("Starting Rerun…")