Add comment related to .complete

Signed-off-by: Kelly A <kellyaa@users.noreply.github.com>
kellyaa · Jun 6, 2024 · c7e00e9 · c7e00e9
1 parent e736417
commit c7e00e9
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/caikit/runtime/train.py b/caikit/runtime/train.py
@@ -313,6 +313,8 @@ def main() -> int:
             future = train(module, wait=True, **train_kwargs)
             info = future.get_info()
             if info.status == TrainingStatus.COMPLETED:
+                # The .complete file indicates all model files have completed
+                # being written to the filesystem
                 complete_path = os.path.join(args.save_path, ".complete")
                 log.info(f"Creating completion file at: {complete_path}")
                 Path(complete_path).touch()