python/gpu_status.py

@@ -1,11 +1,18 @@
 import argparse
-from typing import Protocol
-
 import requests
+
 from renderer import Renderer
 
 parser = argparse.ArgumentParser(description="Prints the status of server nodes and their GPUs.")
 
+
+parser.add_argument(
+    "watch", 
+    type=str,
+    nargs="?",
+    choices=("watch",),
+    help="If the word 'watch' is supplied as first argument, the script will be run in watch mode. Supplying this argument in any other position will have no effect."
+)
 parser.add_argument(
     "-s", 
     "--server", 

python/renderer.py

@@ -60,7 +60,7 @@ class Renderer:
             
             final_lines.extend(lines)            
 
-        final_lines.insert(0, first_line * self.columns)
+        final_lines.insert(0, first_line * min(self.columns, len(line_blocks)))
 
         #lines.append("=" * len(lines[-1]))
 
@@ -86,9 +86,11 @@ class Renderer:
         pad_line = "|" + "-" * (len(lines[-1]) - 2) + "|"
         pad_line_empty = "|" + " " * (len(lines[-1]) - 2) + "|"
 
-        lines.append(pad_line_empty)
+        if self.use_space_lines:
+            lines.append(pad_line_empty)
         lines.append("|" + "=" * (len(lines[-1]) - 2) + "|")
 
+        lines.insert(0, pad_line_empty)
         lines.insert(0, pad_line)
         if self.use_space_lines:
             lines.insert(0, pad_line_empty)
@@ -126,6 +128,8 @@ class Renderer:
 
         if self.display_users:
             lines.append(self.get_rendered_users_line(gpu_dict["running_processes"], len(line_meta)))
+        else:
+            lines.append(empty_line)
 
         if self.use_space_lines:
             lines = [empty_line] + lines