Add ability to have Colab badges on pages

docs/_config.yml

@@ -21,6 +21,10 @@ exclude:
 
 exclude: [vendor]
 
+# This specifies what badges are turned on by default for notebook docs.
+default_badges:
+  colab: true
+
 highlighter: rouge
 markdown: kramdown
 kramdown:

docs/_includes/notebook_colab_link.html

@@ -0,0 +1,5 @@
+<div class="px-2">
+    <a href="https://colab.research.google.com/github/{{site.repository}}/blob/{{page.branch | default: "master"}}/{{page.nb_path}}" target="_blank">
+        <img class="notebook-badge-image" src="{{ "images/colab.svg" | relative_url }}" alt="Open In Colab"/>
+    </a>
+</div>

docs/_layouts/page.html

@@ -5,6 +5,9 @@
 <div class="post-header">
     <a id="{{page.title}}"></a>
     <h1 class="post-title-main">{{ page.title }}</h1>
+    {% if page.nb_path and site.github_username and site.github_repo %}
+        {% unless site.default_badges.colab != true %}{% include notebook_colab_link.html %}{% endunless %}
+    {% endif -%}
 </div>
 
 {% if page.simple_map == true %}

docs/css/customstyles.css

@@ -1333,3 +1333,7 @@ div.input + details {
 .output_wrapper{
     overflow: scroll;
 }
+
+.notebook-badge-image {
+  border:0 !important;
+}

nbdev/templates/jekyll.tpl

@@ -7,6 +7,7 @@ sidebar: home_sidebar
 {% if resources.tags != "" and resources.tags != nil %}tags: {{resources.tags}}{% endif %}
 {% if resources.summary != "" and resources.summary != nil %}summary: "{{resources.summary}}"{% endif %}
 {% if resources.summary != "" and resources.summary != nil %}description: "{{resources.summary}}"{% endif %}
+{% if resources.nb_path != "" and resources.nb_path != nil %}nb_path: "{{resources.nb_path}}"{% endif %}
 ---
 {% include 'autogen.tpl' %}