Attempting to fix formatting.

samples/tools/autogenbench/autogenbench/cli.py

@@ -91,15 +91,11 @@ def main(args=None):
                 sys.stderr.write(help_text + "\n")
                 sys.exit(0)
             else:
-                command["function"](
-                    [invocation_cmd + " " + command["command"]] + args[2:]
-                )
+                command["function"]([invocation_cmd + " " + command["command"]] + args[2:])
                 sys.exit(0)
 
     # Command not found
-    sys.stderr.write(
-        f"Invalid command '{args[1]}'. Available commands include: {commands_list}\n"
-    )
+    sys.stderr.write(f"Invalid command '{args[1]}'. Available commands include: {commands_list}\n")
     sys.exit(2)
 
 

samples/tools/autogenbench/autogenbench/run_cmd.py

@@ -139,9 +139,7 @@ def run_scenarios(
                 print(f"Running scenario {results_repetition}")
 
                 # Expand the scenario
-                expand_scenario(
-                    scenario_dir, instance, results_repetition, requirements
-                )
+                expand_scenario(scenario_dir, instance, results_repetition, requirements)
 
                 # Prepare the environment (keys/values that need to be added)
                 env = get_scenario_env(config_list)
@@ -171,14 +169,10 @@ def expand_scenario(scenario_dir, scenario, output_dir, requirements):
     template = scenario["template"]
 
     # Either key works for finding the substiturions list. "values" may be deprecated in the future
-    substitutions = (
-        scenario["substitutions"] if "substitutions" in scenario else scenario["values"]
-    )
+    substitutions = scenario["substitutions"] if "substitutions" in scenario else scenario["values"]
 
     # Older versions are only one-level deep. Convert them,
-    if len(substitutions) > 0 and isinstance(
-        substitutions[next(iter(substitutions))], str
-    ):
+    if len(substitutions) > 0 and isinstance(substitutions[next(iter(substitutions))], str):
         substitutions = {"scenario.py": substitutions}
 
     copy_operations = []
@@ -217,18 +211,14 @@ def expand_scenario(scenario_dir, scenario, output_dir, requirements):
         else:
             if os.path.isdir(dest_path):
                 # If the destination is a directory, use the same filename
-                shutil.copyfile(
-                    src_path, os.path.join(dest_path, os.path.basename(src_path))
-                )
+                shutil.copyfile(src_path, os.path.join(dest_path, os.path.basename(src_path)))
             else:
                 # Otherwuse use the filename provided
                 shutil.copyfile(src_path, dest_path)
 
     # Copy the requirements file if specified
     if requirements is not None:
-        shutil.copyfile(
-            requirements, pathlib.Path(os.path.join(output_dir, "requirements.txt"))
-        )
+        shutil.copyfile(requirements, pathlib.Path(os.path.join(output_dir, "requirements.txt")))
 
     # Expand templated files
     for templated_file in substitutions.keys():  # Keys are relative file paths
@@ -289,11 +279,7 @@ def run_scenario_natively(work_dir, env, timeout=TASK_TIMEOUT):
 
     # Navigate to the scenario
     os.chdir(work_dir)
-    print(
-        "\n\n"
-        + os.getcwd()
-        + "\n==================================================================="
-    )
+    print("\n\n" + os.getcwd() + "\n===================================================================")
 
     # Prepare the run script
     with open(os.path.join("run.sh"), "wt") as f:
@@ -378,9 +364,7 @@ def run_scenario_in_docker(work_dir, env, timeout=TASK_TIMEOUT, docker_image=Non
         try:
             image = client.images.get(DEFAULT_DOCKER_IMAGE_TAG)
         except docker.errors.ImageNotFound:
-            print(
-                f"Building default Docker image '{DEFAULT_DOCKER_IMAGE_TAG}'. This may take a few minutes..."
-            )
+            print(f"Building default Docker image '{DEFAULT_DOCKER_IMAGE_TAG}'. This may take a few minutes...")
             try:
                 build_default_docker_image(client, DEFAULT_DOCKER_IMAGE_TAG)
                 image = client.images.get(DEFAULT_DOCKER_IMAGE_TAG)
@@ -449,11 +433,7 @@ def run_scenario_in_docker(work_dir, env, timeout=TASK_TIMEOUT, docker_image=Non
 """
         )
 
-    print(
-        "\n\n"
-        + work_dir
-        + "\n==================================================================="
-    )
+    print("\n\n" + work_dir + "\n===================================================================")
 
     # Create and run the container
     abs_path = str(pathlib.Path(work_dir).absolute())
@@ -468,9 +448,7 @@ def run_scenario_in_docker(work_dir, env, timeout=TASK_TIMEOUT, docker_image=Non
     )
 
     # Read the logs in a streaming fashion. Keep an eye on the time to make sure we don't need to stop.
-    docker_timeout = (
-        timeout + 60
-    )  # One full minute after the bash timeout command should have already triggered
+    docker_timeout = timeout + 60  # One full minute after the bash timeout command should have already triggered
     start_time = time.time()
     logs = container.logs(stream=True)
     log_file = open(os.path.join(work_dir, "console_log.txt"), "wt")
@@ -492,9 +470,7 @@ def run_scenario_in_docker(work_dir, env, timeout=TASK_TIMEOUT, docker_image=Non
             # but remember how we got here.
             stopping = True
 
-    if (
-        stopping
-    ):  # By this line we've exited the loop, and the container has actually stopped.
+    if stopping:  # By this line we've exited the loop, and the container has actually stopped.
         log_file.write("\nDocker timed out.\n")
         log_file.flush()
         sys.stdout.write("\nDocker timed out.\n")
@@ -599,16 +575,12 @@ def run_cli(args):
 
     # Don't allow both --docker-image and --native on the same command
     if parsed_args.docker_image is not None and parsed_args.native:
-        sys.exit(
-            "The options --native and --docker-image can not be used together. Exiting."
-        )
+        sys.exit("The options --native and --docker-image can not be used together. Exiting.")
 
     # Warn if running natively
     if parsed_args.native:
         if IS_WIN32:
-            sys.exit(
-                "Running scenarios with --native is not supported in Windows. Exiting."
-            )
+            sys.exit("Running scenarios with --native is not supported in Windows. Exiting.")
 
         if parsed_args.requirements is not None:
             sys.exit("--requirements is not compatible with --native. Exiting.")