autoformat

bdaiinstitute · Sep 16, 2024 · 199698e · 199698e
1 parent c4c978f
commit 199698e
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/predicators/utils.py b/predicators/utils.py
@@ -2579,10 +2579,15 @@ def query_vlm_for_atom_vals(
     # We assume the state.simulator_state contains a list of previous states.
     if "state_history" in state.simulator_state:
         previous_states = state.simulator_state["state_history"]
-    state_imgs_history = [state.simulator_state["images"] for state in previous_states]
+    state_imgs_history = [
+        state.simulator_state["images"] for state in previous_states
+    ]
     vlm_atoms = sorted(vlm_atoms)
     atom_queries_str = [atom.get_vlm_query_str() for atom in vlm_atoms]
-    vlm_query_str, imgs = get_prompt_for_vlm_state_labelling(CFG.vlm_test_time_atom_label_prompt_type, atom_queries_str, state.simulator_state["vlm_atoms_history"], state_imgs_history, [], state.simulator_state["skill_history"])
+    vlm_query_str, imgs = get_prompt_for_vlm_state_labelling(
+        CFG.vlm_test_time_atom_label_prompt_type, atom_queries_str,
+        state.simulator_state["vlm_atoms_history"], state_imgs_history, [],
+        state.simulator_state["skill_history"])
     if vlm is None:
         vlm = create_vlm_by_name(CFG.vlm_model_name)  # pragma: no cover.
     vlm_input_imgs = \