Add notification when user setup task starts

Princeton-CDH · Feb 5, 2025 · 36aa2b4 · 36aa2b4
1 parent 48baa1d
commit 36aa2b4
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/htr2hpc/tasks.py b/src/htr2hpc/tasks.py
@@ -468,7 +468,11 @@ def hpc_user_setup(user_pk=None):
 
     # bash setup script is included with this package
     user_setup_script = settings.HTR2HPC_INSTALL_DIR / "train" / "user_setup.sh"
-
+    user.notify(
+        "Running user setup script, on first run this may take a while...",
+        id="htr2hpc-setup",
+        level="info",
+    )
     try:
         with Connection(
             host=settings.HPC_HOSTNAME,