domain_to_idxs typo

p-lambda · Dec 20, 2023 · cb7b6c6 · cb7b6c6
1 parent 3a1546e
commit cb7b6c6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/experimental/data_selection/dsir_pipeline.py b/experimental/data_selection/dsir_pipeline.py
@@ -218,10 +218,10 @@ def compute_domain_idxs(filter_domains):
             if domain in todo_domains:
                 domain_to_idxs[domain].append(i)
         for domain, idxs in domain_to_idxs.items():
-
             np.save(ds_path / f"{domain.replace(' ', '_')}_idxs.npy", np.asarray(idxs))
-        domain_idxs_path = ds_path / f"{domain.replace(' ', '_')}_idxs.npy"
+
     for domain in filter_domains:
+        domain_idxs_path = ds_path / f"{domain.replace(' ', '_')}_idxs.npy"
         domain_idxs = np.load(domain_idxs_path)
         domain_to_idxs[domain] = domain_idxs