Changes to pass flake8 compatibility.

main.py

@@ -337,14 +337,14 @@
 flags.DEFINE_bool(
     "finer_discretization_at_prev_solution",
     False,
-    "If `True`, the scheduler keeps discretization of 1 around previous solution. (default: False). "
+    "If `True`, the scheduler keeps discretization of 1 around previous solution. "
     "The discretization is dynamically decided based on the occupancy request for "
     "each time slice. (default: False)",
 )
 flags.DEFINE_integer(
     "finer_discretization_window",
     5,
-    "The discretization window around prvious solution that keeps discretization of 1. ",
+    "The window around previous solution that keeps discretization of 1.",
 )
 flags.DEFINE_integer(
     "scheduler_max_time_discretization",
@@ -788,6 +788,7 @@ def main(args):
             _flags=FLAGS,
         )
     elif FLAGS.scheduler == "TetriSched":
+        finer_discretization = FLAGS.finer_discretization_at_prev_solution
         scheduler = TetriSchedScheduler(
             preemptive=FLAGS.preemption,
             runtime=EventTime(FLAGS.scheduler_runtime, EventTime.Unit.US),
@@ -808,7 +809,7 @@ def main(args):
             ),
             dynamic_discretization=FLAGS.scheduler_dynamic_discretization,
             max_occupancy_threshold=FLAGS.scheduler_max_occupancy_threshold,
-            finer_discretization_at_prev_solution=FLAGS.finer_discretization_at_prev_solution,
+            finer_discretization_at_prev_solution=finer_discretization,
             finer_discretization_window=EventTime(
                 FLAGS.finer_discretization_window, EventTime.Unit.US
             ),