fix bug. todo: merge .values and .descendants

make_txt_embedding.py

@@ -39,7 +39,7 @@ def write_txt_features(name_lookup):
     batch_size = args.batch_size // len(openai_imagenet_template)
     for batch, (names, indices) in enumerate(
         tqdm(
-            lib.batched(name_lookup, batch_size),
+            lib.batched(name_lookup.values(), batch_size),
             desc="txt feats",
             total=len(name_lookup) // batch_size,
         )
@@ -84,7 +84,7 @@ def convert_txt_features_to_avgs(name_lookup):
         if rank == "Species":
             continue
         for name, index in tqdm(all_names[i], desc=rank):
-            species = tuple(zip(*((d, i) for d, i in name_lookup.descendants(prefix=name) if len(d) >= 7)))
+            species = tuple(zip(*((d, i) for d, i in name_lookup.descendants(prefix=name) if len(d) >= 6)))
             if not species:
                 logger.warning("No species for %s.", " ".join(name))
                 all_features[:, index] = 0.0