From 0045e649f9bdbe3195530c57bdbe9daa399a9710 Mon Sep 17 00:00:00 2001
From: Ethan Harris <ewah1g13@soton.ac.uk>
Date: Thu, 1 Jul 2021 10:50:33 +0100
Subject: [PATCH] Update docs for new template (#678)

---
 docs/source/_templates/layout.html           | 10 ++++++++++
 docs/source/_templates/theme_variables.jinja |  7 +++++++
 2 files changed, 17 insertions(+)
 create mode 100644 docs/source/_templates/layout.html

diff --git a/docs/source/_templates/layout.html b/docs/source/_templates/layout.html
new file mode 100644
index 0000000000..096cb14155
--- /dev/null
+++ b/docs/source/_templates/layout.html
@@ -0,0 +1,10 @@
+{% extends "!layout.html" %}
+<link rel="canonical" href="{{ theme_canonical_url }}{{ pagename }}.html" />
+
+{% block footer %}
+{{ super() }}
+<script script type="text/javascript">
+  var collapsedSections = ['Callbacks', 'DataModules', 'Datasets', 'DataLoaders', 'Losses', 'Models', 'Vision models', 'Learning Rate Schedulers', 'Data Processing', 'Learning Tasks', 'Community'];
+</script>
+
+{% endblock %}
diff --git a/docs/source/_templates/theme_variables.jinja b/docs/source/_templates/theme_variables.jinja
index 3b0322b120..f47f73d9c4 100644
--- a/docs/source/_templates/theme_variables.jinja
+++ b/docs/source/_templates/theme_variables.jinja
@@ -14,5 +14,12 @@
   'blog': 'https://www.pytorchlightning.ai/blog',
   'resources': 'https://pytorch-lightning.readthedocs.io/en/latest/#community-examples',
   'support': 'https://lightning-bolts.rtfd.io/en/latest/',
+  'community': 'https://pytorch-lightning.slack.com',
+  'forums': 'https://pytorch-lightning.slack.com',
+  'pldocs': 'https://pytorch-lightning.readthedocs.io/en/stable/',
+  'tmdocs': 'https://torchmetrics.readthedocs.io/en/stable/',
+  'fldocs': 'https://lightning-flash.readthedocs.io/en/stable/',
+  'lbdocs': 'https://lightning-bolts.readthedocs.io/en/stable/',
+  'gdocs': 'https://docs.grid.ai/',
 }
 -%}