Run onload callbacks with --warm option

holoviz · Oct 21, 2021 · 47c9240 · 47c9240
1 parent d898ae6
commit 47c9240
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/panel/command/serve.py b/panel/command/serve.py
@@ -24,7 +24,7 @@
 from ..config import config
 from ..io.rest import REST_PROVIDERS
 from ..io.reload import record_modules, watch
-from ..io.server import INDEX_HTML, get_static_routes
+from ..io.server import INDEX_HTML, get_static_routes, set_curdoc
 from ..io.state import state
 
 log = logging.getLogger(__name__)
@@ -227,10 +227,14 @@ def customize_kwargs(self, args, server_kwargs):
                 with record_modules():
                     for app in applications.values():
                         doc = app.create_document()
+                        with set_curdoc(doc):
+                            state._on_load(None)
                         _cleanup_doc(doc)
             else:
                 for app in applications.values():
                     doc = app.create_document()
+                    with set_curdoc(doc):
+                        state._on_load(None)
                     _cleanup_doc(doc)
 
         prefix = args.prefix