vtr_flow: fix python linting

Signed-off-by: Alessandro Comodi <[email protected]>

Author: acomodi

vtr_flow/scripts/python_libs/vtr/flow.py

@@ -13,10 +13,10 @@ class VtrStage(Enum):
         Enum class for the VTR stages\
     """
 
-    odin = 1
-    abc = 2
-    ace = 3
-    vpr = 4
+    ODIN = 1
+    ABC = 2
+    ACE = 3
+    VPR = 4
 
     def __le__(self, other):
         if self.__class__ is other.__class__:
@@ -34,8 +34,8 @@ def run(
     architecture_file,
     circuit_file,
     power_tech_file=None,
-    start_stage=VtrStage.odin,
-    end_stage=VtrStage.vpr,
+    start_stage=VtrStage.ODIN,
+    end_stage=VtrStage.VPR,
     command_runner=vtr.CommandRunner(),
     temp_dir=Path("./temp"),
     odin_args=None,
@@ -175,7 +175,7 @@ def run(
     #
     # RTL Elaboration & Synthesis
     #
-    if should_run_stage(VtrStage.odin, start_stage, end_stage) and circuit_file.suffixes != ".blif":
+    if should_run_stage(VtrStage.ODIN, start_stage, end_stage) and circuit_file.suffixes != ".blif":
         vtr.odin.run(
             architecture_copy,
             next_stage_netlist,
@@ -195,7 +195,7 @@ def run(
     #
     # Logic Optimization & Technology Mapping
     #
-    if should_run_stage(VtrStage.abc, start_stage, end_stage):
+    if should_run_stage(VtrStage.ABC, start_stage, end_stage):
         vtr.abc.run(
             architecture_copy,
             next_stage_netlist,
@@ -216,7 +216,7 @@ def run(
     if power_tech_file:
         # The user provided a tech file, so do power analysis
 
-        if should_run_stage(VtrStage.ace, start_stage, end_stage):
+        if should_run_stage(VtrStage.ACE, start_stage, end_stage):
             vtr.ace.run(
                 next_stage_netlist,
                 old_netlist=post_odin_netlist,
@@ -241,7 +241,7 @@ def run(
     #
     # Pack/Place/Route
     #
-    if should_run_stage(VtrStage.vpr, start_stage, end_stage):
+    if should_run_stage(VtrStage.VPR, start_stage, end_stage):
         # Copy the input netlist for input to vpr
         shutil.copyfile(str(next_stage_netlist), str(pre_vpr_netlist))
         route_fixed_w = "route_chan_width" in vpr_args

vtr_flow/scripts/python_libs/vtr/parse_vtr_task.py

@@ -566,5 +566,5 @@ def find_latest_run_dir(config):
 
 
 if __name__ == "__main__":
-    retval = vtr_command_main(sys.argv[1:])
-    sys.exit(retval)
+    RETVAL = vtr_command_main(sys.argv[1:])
+    sys.exit(RETVAL)

vtr_flow/scripts/run_vtr_flow.py

@@ -19,6 +19,8 @@
 
 BASIC_VERBOSITY = 1
 
+VTR_STAGES = ["odin", "abc", "ace", "vpr"]
+
 # pylint: disable=too-few-public-methods
 class VtrStageArgparseAction(argparse.Action):
     """
@@ -27,13 +29,13 @@ class VtrStageArgparseAction(argparse.Action):
 
     def __call__(self, parser, namespace, value, option_string=None):
         if value == "odin":
-            setattr(namespace, self.dest, vtr.VtrStage.odin)
+            setattr(namespace, self.dest, vtr.VtrStage.ODIN)
         elif value == "abc":
-            setattr(namespace, self.dest, vtr.VtrStage.abc)
+            setattr(namespace, self.dest, vtr.VtrStage.ABC)
         elif value == "vpr":
-            setattr(namespace, self.dest, vtr.VtrStage.vpr)
+            setattr(namespace, self.dest, vtr.VtrStage.VPR)
         elif value == "lec":
-            setattr(namespace, self.dest, vtr.VtrStage.lec)
+            setattr(namespace, self.dest, vtr.VtrStage.LEC)
         else:
             raise argparse.ArgumentError(self, "Invalid VTR stage '" + value + "'")
 
@@ -107,8 +109,8 @@ def vtr_command_argparser(prog=None):
     parser.add_argument(
         "-start",
         "-starting_stage",
-        choices=str(list(vtr.VtrStage)),
-        default=vtr.VtrStage.odin,
+        choices=VTR_STAGES,
+        default=vtr.VtrStage.ODIN,
         action=VtrStageArgparseAction,
         help="Starting stage of the VTR flow.",
     )
@@ -129,8 +131,8 @@ def vtr_command_argparser(prog=None):
     parser.add_argument(
         "-end",
         "-ending_stage",
-        choices=str(list(vtr.VtrStage)),
-        default=vtr.VtrStage.vpr,
+        choices=VTR_STAGES,
+        default=vtr.VtrStage.VPR,
         action=VtrStageArgparseAction,
         help="Ending stage of the VTR flow.",
     )