Enzo-py commited on
Commit
06ca312
·
verified ·
1 Parent(s): 63276fa

Deploy 14f9835 from GitHub

Browse files
Files changed (1) hide show
  1. app.py +11 -0
app.py CHANGED
@@ -65,6 +65,17 @@ def load_stimuli():
65
  or f.split("/")[-1].startswith(DATASET_SPLIT))]
66
  if not files:
67
  raise RuntimeError(f"No parquet file for split '{DATASET_SPLIT}' in {DATASET_REPO}@{sha[:8]}")
 
 
 
 
 
 
 
 
 
 
 
68
  local = Path(snapshot_download(DATASET_REPO, repo_type="dataset", revision=sha,
69
  allow_patterns=files, token=HF_TOKEN))
70
  STIM_DIR.mkdir(parents=True, exist_ok=True)
 
65
  or f.split("/")[-1].startswith(DATASET_SPLIT))]
66
  if not files:
67
  raise RuntimeError(f"No parquet file for split '{DATASET_SPLIT}' in {DATASET_REPO}@{sha[:8]}")
68
+ # The same parquet file uploaded at two paths (e.g. root and data/) would duplicate every row.
69
+ seen, unique = set(), []
70
+ for info in sorted(api.get_paths_info(DATASET_REPO, files, repo_type="dataset", revision=sha),
71
+ key=lambda i: i.path):
72
+ oid = info.lfs.sha256 if info.lfs else info.blob_id
73
+ if oid in seen:
74
+ print(f"Skipping {info.path}: identical to an already selected file")
75
+ continue
76
+ seen.add(oid)
77
+ unique.append(info.path)
78
+ files = unique
79
  local = Path(snapshot_download(DATASET_REPO, repo_type="dataset", revision=sha,
80
  allow_patterns=files, token=HF_TOKEN))
81
  STIM_DIR.mkdir(parents=True, exist_ok=True)