🧹 fmt fixes for launch scripts

Signed-off-by: Kelly A <[email protected]>
kellyaa · May 6, 2024 · 75dbbf1 · 75dbbf1
1 parent e38d859
commit 75dbbf1
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 8 deletions.
diff --git a/build/accelerate_launch.py b/build/accelerate_launch.py
@@ -55,14 +55,14 @@ def main():
             )
         )
         sys.exit(1)
-    except Exception as e: # pylint: disable=broad-except
+    except Exception as e:  # pylint: disable=broad-except
         logging.error(traceback.format_exc())
         write_termination_log("Unhandled exception during training")
         sys.exit(200)
 
     try:
         launch_command(args)
-    except Exception as e: # pylint: disable=broad-except
+    except Exception as e:  # pylint: disable=broad-except
         logging.error(traceback.format_exc)
         write_termination_log("Unhandled exception during training")
         sys.exit(200)

diff --git a/build/launch_training.py b/build/launch_training.py
@@ -69,7 +69,7 @@ def main():
             tune_config,
             merge_model,
         ) = process_launch_training_args(job_config)
-    except Exception as e: # pylint: disable=broad-except
+    except Exception as e:  # pylint: disable=broad-except
         logging.error(traceback.format_exc())
         write_termination_log(
             "Exception raised during training. This may be a problem with your input: {}".format(
@@ -97,7 +97,7 @@ def main():
                 f"Exception raised during training. This may be a problem with your input: {e}"
             )
             sys.exit(1)
-        except Exception as e: # pylint: disable=broad-except
+        except Exception as e:  # pylint: disable=broad-except
             logging.error(traceback.format_exc())
             write_termination_log("Unhandled exception during training")
             sys.exit(200)
@@ -126,7 +126,7 @@ def main():
                     base_model=model_args.model_name_or_path,
                     save_tokenizer=True,
                 )
-            except Exception as e: # pylint: disable=broad-except
+            except Exception as e:  # pylint: disable=broad-except
                 logging.error(traceback.format_exc())
                 write_termination_log("Exception encountered merging model checkpoints")
                 sys.exit(200)
@@ -144,7 +144,7 @@ def main():
                     original_output_dir,
                     dirs_exist_ok=True,
                 )
-            except Exception as e: # pylint: disable=broad-except
+            except Exception as e:  # pylint: disable=broad-except
                 logging.error(traceback.format_exc())
                 write_termination_log(
                     "Exception encountered writing output model to storage"
@@ -158,7 +158,7 @@ def main():
             )
             if os.path.exists(train_logs_filepath):
                 shutil.copy(train_logs_filepath, original_output_dir)
-        except Exception as e: # pylint: disable=broad-except
+        except Exception as e:  # pylint: disable=broad-except
             logging.error(traceback.format_exc())
             # Continue, don't fail the training because of this
 

diff --git a/build/utils.py b/build/utils.py
@@ -32,7 +32,7 @@ def write_termination_log(text, log_file="/dev/termination-log"):
     try:
         with open(log_file, "a", encoding="utf-8") as handle:
             handle.write(text)
-    except Exception as e: # pylint: disable=broad-except
+    except Exception as e:  # pylint: disable=broad-except
         logging.warning("Unable to write termination log due to error {}".format(e))