fix formatting

Author: david-zlai

api/py/ai/chronon/repo/run.py

@@ -214,7 +214,8 @@ def download_jar(
         spark_version="2.4.0",
         skip_download=False,
 ):
-    assert (spark_version in SUPPORTED_SPARK), f"Received unsupported spark version {spark_version}. Supported spark versions are {SUPPORTED_SPARK}"
+    assert (spark_version in SUPPORTED_SPARK), (f"Received unsupported spark version {spark_version}. "
+                                                f"Supported spark versions are {SUPPORTED_SPARK}")
     scala_version = SCALA_VERSION_FOR_SPARK[spark_version]
     maven_url_prefix = os.environ.get("CHRONON_MAVEN_MIRROR_PREFIX", None)
     default_url_prefix = (
@@ -429,9 +430,10 @@ def __init__(self, args, jar_path):
                 raise e
             possible_modes = list(
                 ROUTES[self.conf_type].keys()) + UNIVERSAL_ROUTES
-            assert (args["mode"] in possible_modes), "Invalid mode:{} for conf:{} of type:{}, please choose from {}".format(
-                args["mode"], self.conf, self.conf_type, possible_modes
-            )
+            assert (
+                    args["mode"] in possible_modes), ("Invalid mode:{} for conf:{} of type:{}, please choose from {}"
+                                                      .format(args["mode"], self.conf, self.conf_type, possible_modes
+                                                              ))
         else:
             self.conf_type = args["conf_type"]
         self.ds = args["end_ds"] if "end_ds" in args and args["end_ds"] else args["ds"]
@@ -550,8 +552,8 @@ def run(self):
                         for start_ds, end_ds in date_ranges:
                             if not self.dataproc:
                                 command = (
-                                    "bash {script} --class ai.chronon.spark.Driver " +
-                                    "{jar} {subcommand} {args} {additional_args}"
+                                        "bash {script} --class ai.chronon.spark.Driver " +
+                                        "{jar} {subcommand} {args} {additional_args}"
                                 ).format(
                                     script=self.spark_submit,
                                     jar=self.jar_path,
@@ -598,8 +600,8 @@ def run(self):
                     else:
                         if not self.dataproc:
                             command = (
-                                "bash {script} --class ai.chronon.spark.Driver "
-                                + "{jar} {subcommand} {args} {additional_args}"
+                                    "bash {script} --class ai.chronon.spark.Driver "
+                                    + "{jar} {subcommand} {args} {additional_args}"
                             ).format(
                                 script=self.spark_submit,
                                 jar=self.jar_path,
@@ -662,8 +664,7 @@ def _gen_final_args(self, start_ds=None, end_ds=None, override_conf_path=None):
             " --start-partition-override=" + start_ds if start_ds else ""
         )
 
-        final_args = base_args + " " + \
-                     str(self.args) + override_start_partition_arg
+        final_args = base_args + " " + str(self.args) + override_start_partition_arg
 
         return final_args