Rename vars for clarity

app.py

@@ -33,11 +33,11 @@ def html_template(rrd: str, app_url: str = "https://app.rerun.io") -> str:
  return f"""<div style="width:100%; height:70vh;"><iframe style="width:100%; height:100%;" src="{app_url}?url={encoded_url}" frameborder="0" allowfullscreen=""></iframe></div>"""
 
 
-def show_dataset(dataset_id: str, episode_id: int) -> str:
+def show_dataset(dataset_id: str, episode_index: int) -> str:
  rr.init("dataset")
 
  # TODO(jleibs): manage cache better and put in proper storage
- filename = Path(f"tmp/{dataset_id}_{episode_id}.rrd")
+ filename = Path(f"tmp/{dataset_id}_{episode_index}.rrd")
  if not filename.exists():
  filename.parent.mkdir(parents=True, exist_ok=True)
 
@@ -46,7 +46,9 @@ def show_dataset(dataset_id: str, episode_id: int) -> str:
  dataset = load_dataset(dataset_id, split="train", streaming=True)
 
  # This is for LeRobot datasets (https://huggingface.co/lerobot):
- ds_subset = dataset.filter(lambda frame: "episode_index" not in frame or frame["episode_index"] == episode_id)
+ ds_subset = dataset.filter(
+ lambda frame: "episode_index" not in frame or frame["episode_index"] == episode_index
+ )
 
  log_dataset_to_rerun(ds_subset)
 
@@ -61,14 +63,14 @@ with gr.Blocks() as demo:
  placeholder="Search for models on Huggingface",
  search_type="dataset",
  )
- episode_id = gr.Number(1, label="Episode ID")
+ episode_index = gr.Number(1, label="Episode Index")
  button = gr.Button("Show Dataset")
  with gr.Row():
  rrd = gr.File()
  with gr.Row():
  viewer = gr.HTML()
 
- button.click(show_dataset, inputs=[search_in, episode_id], outputs=rrd)
+ button.click(show_dataset, inputs=[search_in, episode_index], outputs=rrd)
  rrd.change(
  html_template,
  js="""(rrd) => { console.log(rrd.url); return rrd.url}""",

main.py

@@ -20,14 +20,16 @@ def main() -> None:
 
  parser = argparse.ArgumentParser(description="Log a HuggingFace dataset to Rerun.")
  parser.add_argument("--dataset", default="lerobot/pusht", help="The name of the dataset to load")
- parser.add_argument("--episode-id", default=1, help="Which episode to select")
+ parser.add_argument("--episode-index", default=1, help="Which episode to select")
  args = parser.parse_args()
 
  print("Loading dataset…")
  dataset = load_dataset(args.dataset, split="train", streaming=True)
 
  # This is for LeRobot datasets (https://huggingface.co/lerobot):
- ds_subset = dataset.filter(lambda frame: "episode_index" not in frame or frame["episode_index"] == args.episode_id)
+ ds_subset = dataset.filter(
+ lambda frame: "episode_index" not in frame or frame["episode_index"] == args.episode_index
+ )
 
  print("Starting Rerun…")
  rr.init(f"rerun_example_huggingface {args.dataset}", spawn=True)