black version update

Author: TLSDC

requirements.txt

@@ -1,4 +1,4 @@
-black[jupyter]>=24.2.0
+black[jupyter]>=24.2.0,<25
 blacken-docs
 pre-commit
 pytest==7.3.2

src/agentlab/analyze/error_analysis/summarizer.py

@@ -39,16 +39,18 @@ def summarize(self, obs: StepInfo, next_obs: StepInfo, past_summaries: list[str]
         if self.use_diff:
             next_obs_message = _diff(obs_message, next_obs_message)
 
-        return self.parse(self.llm(
-            self.make_prompt(
-                obs_message,
-                action,
-                next_obs_message,
-                past_summaries,
-                goal,
-                obs.obs.get("plan", "No plan available"),
-            )
-        )['content'])
+        return self.parse(
+            self.llm(
+                self.make_prompt(
+                    obs_message,
+                    action,
+                    next_obs_message,
+                    past_summaries,
+                    goal,
+                    obs.obs.get("plan", "No plan available"),
+                )
+            )["content"]
+        )
 
     def make_prompt(
         self, past_obs_message, action, current_obs_message, past_summaries, goal, plan
@@ -64,7 +66,9 @@ def make_prompt(
         )
 
     def parse(self, raw_output: str) -> dict:
-        parsed_result = parse_html_tags(raw_output, keys=["changeSummary", "actionAssessment", "explanation", "suggestion"])[0]
+        parsed_result = parse_html_tags(
+            raw_output, keys=["changeSummary", "actionAssessment", "explanation", "suggestion"]
+        )[0]
         return parsed_result
 
 
@@ -105,15 +109,16 @@ def make_change_summaries(self, exp_result: ExpResult) -> list[str]:
         # it is generally the case, but exps can sometimes fail in a weird way and not save the last step_info
         # TODO:(thibault) make some checks or w/e
         for step, next_step in zip(exp_result.steps_info[:-1], exp_result.steps_info[1:]):
-            summaries.append(
-                self.change_summarizer.summarize(step, next_step, summaries)
-            )
+            summaries.append(self.change_summarizer.summarize(step, next_step, summaries))
         return summaries
 
     def parse(self, raw_output: str) -> dict:
-        parsed_result = parse_html_tags(raw_output, keys=["explanation", "success", "errorCategory"])[0]
+        parsed_result = parse_html_tags(
+            raw_output, keys=["explanation", "success", "errorCategory"]
+        )[0]
         return parsed_result
 
+
 @dataclass
 class EpisodeErrorSummarizer(EpisodeSummarizer):
 
@@ -124,7 +129,7 @@ def make_prompt(self, exp_results: ExpResult, summaries: list[str]):
         goal = exp_results.steps_info[0].obs["goal"]
 
         def format_summary(summary):
-            res = ''
+            res = ""
             for key, value in summary.items():
                 res += f"{key}: {value}\n"
             return res