Fix wandb alert title overflow, fix show_gpu_table account names

tatsuya-ishihara · Oct 18, 2024 · 2bfad93 · 2bfad93
1 parent f6c312d
commit 2bfad93
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/infinigen/datagen/manage_jobs.py b/infinigen/datagen/manage_jobs.py
@@ -679,7 +679,7 @@ def manage_datagen_jobs(all_scenes, elapsed, num_concurrent, disk_sleep_threshol
         print(message)
         if wandb is not None:
             wandb.alert(
-                title=f"{args.output_folder} full",
+                title=f"{args.output_folder.name} sleeping for full disk",
                 text=message,
                 wait_duration=3 * 60 * 60,
             )

diff --git a/infinigen/datagen/util/show_gpu_table.py b/infinigen/datagen/util/show_gpu_table.py
@@ -35,7 +35,7 @@ def get_gpu_nodes():
     for line in sinfo_output.splitlines():
         node, group, gres, totalmem = line.split()
         if group != "all":
-            if group in {"pvl", "cs*"}:
+            if group in {"pvl", "cs*", "pnlp"}:
                 num_cpus = int(cpu_regex(gres).group(1))
                 shared_node_mem[node] = int((int(totalmem) / 1024) / num_cpus)
             if gres_regex(gres):