make format-py

Author: soheilshahrouz

vtr_flow/scripts/download_noc_mlp.py

@@ -17,6 +17,7 @@
 import shutil
 import errno
 
+
 class ExtractionError(Exception):
     """
     Raised when extracting the downlaoded file fails
@@ -39,8 +40,9 @@ def parse_args():
                     does nothing (unless --force is specified).
                   """
     )
-    parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter,
-                                     description=description)
+    parser = argparse.ArgumentParser(
+        formatter_class=argparse.ArgumentDefaultsHelpFormatter, description=description
+    )
 
     parser.add_argument(
         "--vtr_flow_dir",
@@ -132,7 +134,7 @@ def extract_to_vtr_flow_dir(args, tar_gz_filename):
                 raise ExtractionError("{} should be a directory".format(directory))
 
     # Create a temporary working directory
-    tmpdir = tempfile.mkdtemp(suffix="download_NoC_MLP", dir= os.path.abspath("."))
+    tmpdir = tempfile.mkdtemp(suffix="download_NoC_MLP", dir=os.path.abspath("."))
     try:
         # Extract the contents of the .tar.gz archive directly into the destination directory
         with tarfile.open(tar_gz_filename, "r:gz") as tar:

vtr_flow/scripts/download_titan.py

@@ -41,8 +41,9 @@ def parse_args():
                     does nothing (unless --force is specified).
                   """
     )
-    parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter,
-                                     description=description)
+    parser = argparse.ArgumentParser(
+        formatter_class=argparse.ArgumentDefaultsHelpFormatter, description=description
+    )
 
     parser.add_argument(
         "--titan_version", default="2.0.0", help="Titan release version to download"

vtr_flow/scripts/python_libs/vtr/log_parse.py

@@ -25,7 +25,7 @@ def __init__(self, name, filename, regex_str, default_value=None):
         self._filename = filename
         # Look for the specified pattern somewhere in the line, but any characters
         # can occur before and after it. Detailed in GitHub Issue #2743.
-        self._regex = re.compile(f'^.*{regex_str}.*$')
+        self._regex = re.compile(f"^.*{regex_str}.*$")
         self._default_value = default_value
 
     def name(self):

vtr_flow/scripts/python_libs/vtr/task.py

@@ -55,7 +55,7 @@ def __init__(
         pad_file=None,
         additional_files=None,
         additional_files_list_add=None,
-        circuit_constraint_list_add=None
+        circuit_constraint_list_add=None,
     ):
         self.task_name = task_name
         self.config_dir = config_dir
@@ -82,9 +82,9 @@ def __init__(
         self.pad_file = pad_file
         self.additional_files = additional_files
         self.additional_files_list_add = additional_files_list_add
-        self.circuit_constraints = parse_circuit_constraint_list(circuit_constraint_list_add,
-                                                                 self.circuits,
-                                                                 self.archs)
+        self.circuit_constraints = parse_circuit_constraint_list(
+            circuit_constraint_list_add, self.circuits, self.archs
+        )
 
 
 # pylint: enable=too-few-public-methods
@@ -306,9 +306,8 @@ def check_include_fields(config_file, key_values):
                 )
             )
 
-def parse_circuit_constraint_list(
-    circuit_constraint_list, circuits_list, arch_list
-) -> dict:
+
+def parse_circuit_constraint_list(circuit_constraint_list, circuits_list, arch_list) -> dict:
     """
     Parse the circuit constraints passed in via the config file.
     Circuit constraints are expected to have the following syntax:
@@ -342,38 +341,39 @@ def parse_circuit_constraint_list(
     # Parse the circuit constraint list
     for circuit_constraint in circuit_constraint_list:
         # Remove the round brackets.
-        if circuit_constraint[0] != '(' or circuit_constraint[-1] != ')':
-            raise VtrError(f"Circuit constraint syntax error: \"{circuit_constraint}\"")
+        if circuit_constraint[0] != "(" or circuit_constraint[-1] != ")":
+            raise VtrError(f'Circuit constraint syntax error: "{circuit_constraint}"')
         circuit_constraint = circuit_constraint[1:-1]
         # Split the circuit and the constraint
-        split_constraint_line = circuit_constraint.split(',')
+        split_constraint_line = circuit_constraint.split(",")
         if len(split_constraint_line) != 2:
-            raise VtrError(f"Circuit constraint has too many arguments: \"{circuit_constraint}\"")
+            raise VtrError(f'Circuit constraint has too many arguments: "{circuit_constraint}"')
         circuit = split_constraint_line[0].strip()
         constraint = split_constraint_line[1].strip()
         # Check that the circuit actually exists.
         if circuit not in circuits_list:
-            raise VtrError(f"Cannot constrain circuit \"{circuit}\", circuit has not been added")
+            raise VtrError(f'Cannot constrain circuit "{circuit}", circuit has not been added')
         # Parse the constraint
         split_constraint = constraint.split("=")
         if len(split_constraint) != 2:
-            raise VtrError(f"Circuit constraint syntax error: \"{circuit_constraint}\"")
+            raise VtrError(f'Circuit constraint syntax error: "{circuit_constraint}"')
         constr_key = split_constraint[0].strip()
         constr_val = split_constraint[1].strip()
         # Check that the constr_key is valid.
         if constr_key not in circuit_constraint_keys:
-            raise VtrError(f"Invalid constraint \"{constr_key}\" used on circuit \"{circuit}\"")
+            raise VtrError(f'Invalid constraint "{constr_key}" used on circuit "{circuit}"')
         # In the case of arch constraints, make sure this arch exists.
         if constr_key == "arch" and constr_val not in arch_list:
-            raise VtrError(f"Cannot constrain arch \"{constr_key}\", arch has not been added")
+            raise VtrError(f'Cannot constrain arch "{constr_key}", arch has not been added')
         # Make sure this circuit is not already constrained with this constr_arg
         if res_circuit_constraints[circuit][constr_key] is not None:
-            raise VtrError(f"Circuit \"{circuit}\" cannot be constrained more than once")
+            raise VtrError(f'Circuit "{circuit}" cannot be constrained more than once')
         # Add the constraint for this circuit
         res_circuit_constraints[circuit][constr_key] = constr_val
 
     return res_circuit_constraints
 
+
 def shorten_task_names(configs, common_task_prefix):
     """
     Shorten the task names of the configs by remove the common task prefix.
@@ -582,10 +582,7 @@ def create_jobs(args, configs, after_run=False) -> List[Job]:
             work_dir = get_work_dir_addr(arch, circuit, noc_traffic)
 
             run_dir = (
-                str(
-                    Path(get_latest_run_dir(find_task_dir(config, args.alt_tasks_dir)))
-                    / work_dir
-                )
+                str(Path(get_latest_run_dir(find_task_dir(config, args.alt_tasks_dir))) / work_dir)
                 if after_run
                 else str(
                     Path(get_next_run_dir(find_task_dir(config, args.alt_tasks_dir))) / work_dir
@@ -779,6 +776,7 @@ def ret_expected_vpr_status(arch, circuit, golden_results, script_params=None):
 
     return golden_metrics["vpr_status"]
 
+
 def apply_cmd_line_circuit_constraints(cmd, circuit, config):
     """
     Apply the circuit constraints to the command line. If the circuit is not
@@ -793,6 +791,7 @@ def apply_cmd_line_circuit_constraints(cmd, circuit, config):
     if circuit_vpr_constraints is not None:
         cmd += ["--read_vpr_constraints", circuit_vpr_constraints]
 
+
 def resolve_vtr_source_file(config, filename, base_dir=""):
     """
     Resolves an filename with a base_dir

vtr_flow/scripts/python_libs/vtr/util.py

@@ -154,15 +154,15 @@ def run_system_command(
             # or RR graph files to VPR. PWD environment variable is updated
             # manually to prevent capnproto from throwing exceptions.
             modified_environ = os.environ.copy()
-            modified_environ['PWD'] = str(temp_dir)
+            modified_environ["PWD"] = str(temp_dir)
 
             proc = subprocess.Popen(
                 cmd,
                 stdout=subprocess.PIPE,  # We grab stdout
                 stderr=stderr,  # stderr redirected to stderr
                 universal_newlines=True,  # Lines always end in \n
                 cwd=str(temp_dir),  # Where to run the command
-                env=modified_environ
+                env=modified_environ,
             )
 
             # Read the output line-by-line and log it

vtr_flow/scripts/tuning_runs/control_runs.py

@@ -46,9 +46,7 @@ def parse_results(input_path):
     Parse the output results
     """
     # Find the runXXX directory with the largest XXX
-    run_dirs = [
-        d for d in os.listdir(input_path) if d.startswith("run") and d[3:].isdigit()
-    ]
+    run_dirs = [d for d in os.listdir(input_path) if d.startswith("run") and d[3:].isdigit()]
     if not run_dirs:
         print("No runXXX directories found in the specified input path.")
         sys.exit(1)
@@ -63,9 +61,9 @@ def parse_results(input_path):
         sys.exit(1)
 
     # Read the parse_results.txt file and write to full_res.csv
-    with open(
-        os.path.join(largest_run_path, "parse_results.txt"), "r"
-    ) as txt_file, open(full_res_csv_path, "w", newline="") as csv_file:
+    with open(os.path.join(largest_run_path, "parse_results.txt"), "r") as txt_file, open(
+        full_res_csv_path, "w", newline=""
+    ) as csv_file:
         reader = csv.reader(txt_file, delimiter="\t")
         writer = csv.writer(csv_file)
 
@@ -97,9 +95,7 @@ def parse_results(input_path):
     print("Generated average seed results")
 
     # Generate gmean_res.csv
-    generate_geomean_res_csv(
-        os.path.join(largest_run_path, "avg_seed.csv"), largest_run_path
-    )
+    generate_geomean_res_csv(os.path.join(largest_run_path, "avg_seed.csv"), largest_run_path)
     print("Generated geometric average results over all the circuits")
 
     generate_xlsx(largest_run_path)
@@ -147,8 +143,7 @@ def parse_script_params(script_params):
                 j = i + key_length
 
                 while j < len(parts) and not any(
-                    parts[j : j + len(k.split("_"))] == k.split("_")
-                    for k in PARAMS_DICT
+                    parts[j : j + len(k.split("_"))] == k.split("_") for k in PARAMS_DICT
                 ):
                     value_parts.append(parts[j])
                     j += 1