fix: also generate testB

scripts/preprocess_data/cogent/preprocess.py

@@ -91,7 +91,7 @@ def generate_masks(use_cuda: bool, module_info: DotDict):
     logging.info("Loading Detectron...")
     detectron: DefaultPredictor = load_detectron(use_cuda, module_info)
 
-    for dataset in ["trainA", "valA", "testA"]:
+    for dataset in ["trainA", "valA", "testA", "testB"]:
         # Load the scenes
         logging.info("Loading scenes...")
         path = f"data/cogent/raw_scenes/CLEVR_{dataset}_splitscenes.json"
@@ -146,7 +146,7 @@ def match_masks() -> None:
         in a dataset of millions of objects.
     """
     os.makedirs("data/cogent/scenes", exist_ok=True)
-    for dataset in ["trainA", "valA", "testA"]:
+    for dataset in ["trainA", "valA", "testA", "testB"]:
 
         logging.info("Loading masks...")
         masks = load_json(f"data/cogent/raw_scenes/CLEVR_{dataset}_masks.json")