Emil Ernerfeldt commited on
Commit
d63b041
·
1 Parent(s): 147aaec
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -71,7 +71,7 @@ def main():
71
  print("Loading dataset…")
72
  dataset = load_dataset(args.dataset, split="train")
73
 
74
- print("Selecting episode {args.episode_id}…")
75
  ds_subset = dataset.filter(lambda frame: frame["episode_id"] == args.episode_id)
76
 
77
  print("Starting Rerun…")
 
71
  print("Loading dataset…")
72
  dataset = load_dataset(args.dataset, split="train")
73
 
74
+ print(f"Selecting episode {args.episode_id}…")
75
  ds_subset = dataset.filter(lambda frame: frame["episode_id"] == args.episode_id)
76
 
77
  print("Starting Rerun…")