headers

Signed-off-by: oliver könig <okoenig@nvidia.com>

Author: ko3n1g

nemo_run/core/execution/dgxcloud.py

@@ -347,10 +347,10 @@ def status(self, job_id: str) -> Optional[DGXCloudState]:
         r_json = response.json()
         return DGXCloudState(r_json["phase"])
 
-    def _stream_url_sync(self, url: str, q: queue.Queue):
+    def _stream_url_sync(self, url: str, headers: Dict[str, str], q: queue.Queue):
         """Stream a single URL using requests and put chunks into the queue"""
         try:
-            with requests.get(url, stream=True, verify=False) as response:
+            with requests.get(url, stream=True, headers=headers, verify=False) as response:
                 for line in response.iter_lines(decode_unicode=True):
                     q.put(line)
         except Exception as e:
@@ -400,7 +400,12 @@ def fetch_logs(
         active_urls = set(urls)
 
         # Start threads
-        threads = [threading.Thread(target=self._stream_url_sync, args=(url, q)) for url in urls]
+        threads = [
+            threading.Thread(
+                target=self._stream_url_sync, args=(url, self._default_headers(token=token), q)
+            )
+            for url in urls
+        ]
         for t in threads:
             t.start()