test

Author: atharva-2001

tardis/io/logger/logger.py

@@ -175,7 +175,7 @@ def finalize_widget_logging(self):
         """Finalize widget logging by embedding the final state.
         """
         # Embed the final state for Jupyter environments
-        if (Environment.is_notebook() and hasattr(self, 'display_handles') 
+        if (Environment.allows_widget_display() and hasattr(self, 'display_handles') 
             and hasattr(self, 'display_ids') and self.display_handles and self.display_ids):
             print("Embedding the final state for Jupyter environments")
             for level, column in self.log_columns.items():
@@ -255,7 +255,7 @@ def logging_state(log_level, tardis_config, specific_log_level=None, display_log
     tardislogger.configure_logging(log_level, tardis_config, specific_log_level)
     use_widget = display_logging_widget and Environment.allows_widget_display()
 
-    if Environment.is_notebook() or Environment.is_sshjh():
+    if Environment.is_notebook() or Environment.is_sshjh() or Environment.is_sphinx():
         display_handles = {}
         display_ids = {}
         for level, column in log_columns.items():

tardis/workflows/simple_tardis_workflow.py

@@ -26,7 +26,7 @@
 
 
 class SimpleTARDISWorkflow(WorkflowLogging):
-    show_progress_bars = Environment.is_notebook()
+    show_progress_bars = Environment.allows_widget_display()
     enable_virtual_packet_logging = False
     log_level = None
     specific_log_level = None