From d9f6426e0da555d42ad3017850a52dad8e5f7749 Mon Sep 17 00:00:00 2001
From: Thomas Wang <24695242+thomasw21@users.noreply.github.com>
Date: Tue, 1 Nov 2022 00:45:20 +0100
Subject: [PATCH] Check main thread when always rendering. (#865)

Check main thread
---
 nerfstudio/engine/trainer.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/nerfstudio/engine/trainer.py b/nerfstudio/engine/trainer.py
index a3162ff32..8959b27c7 100644
--- a/nerfstudio/engine/trainer.py
+++ b/nerfstudio/engine/trainer.py
@@ -174,12 +174,14 @@ class Trainer:
             self.save_checkpoint(step)
             self._always_render(step)
 
+    @check_main_thread
     def _always_render(self, step):
         if self.config.is_viewer_enabled():
             while True:
                 self.viewer_state.vis["renderingState/isTraining"].write(False)
                 self._update_viewer_state(step)
 
+    @check_main_thread
     def _check_viewer_warnings(self) -> None:
         """Helper to print out any warnings regarding the way the viewer/loggers are enabled"""
         if self.config.is_viewer_enabled():