Spaces:
Sleeping
Sleeping
Emil Ernerfeldt
commited on
Commit
·
d63b041
1
Parent(s):
147aaec
Fix typo
Browse files
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…")
|