prompt: We are currently solving the following issue within our repository. Here is the issue text:
  --- BEGIN ISSUE ---
  Open in Colab URL broken in 0.1.1
  The latest 0.1.1 release broke the "Open in Colab" URL in our tutorials. Still trying to fix this.
  
  --- END ISSUE ---
  
  Below are some code segments, each from a relevant file. One or more of these files may contain bugs.
  --- BEGIN FILES ---
  Path: `docs/conf.py`
  Content:
  ```
  1 # Copyright (c) Microsoft Corporation. All rights reserved.
  2 # Licensed under the MIT License.
  3 
  4 # Configuration file for the Sphinx documentation builder.
  5 #
  6 # This file only contains a selection of the most common options. For a full
  7 # list see the documentation:
  8 # https://www.sphinx-doc.org/en/master/usage/configuration.html
  9 
  10 # -- Path setup --------------------------------------------------------------
  11 
  12 import os
  13 import sys
  14 
  15 import pytorch_sphinx_theme
  16 
  17 # If extensions (or modules to document with autodoc) are in another directory,
  18 # add these directories to sys.path here. If the directory is relative to the
  19 # documentation root, use os.path.abspath to make it absolute, like shown here.
  20 sys.path.insert(0, os.path.abspath(".."))
  21 
  22 import torchgeo  # noqa: E402
  23 
  24 # -- Project information -----------------------------------------------------
  25 
  26 project = "torchgeo"
  27 copyright = "2021, Microsoft Corporation"
  28 author = torchgeo.__author__
  29 version = ".".join(torchgeo.__version__.split(".")[:2])
  30 release = torchgeo.__version__
  31 
  32 
  33 # -- General configuration ---------------------------------------------------
  34 
  35 # Add any Sphinx extension module names here, as strings. They can be
  36 # extensions coming with Sphinx (named 'sphinx.ext.*') or your custom
  37 # ones.
  38 extensions = [
  39     "sphinx.ext.autodoc",
  40     "sphinx.ext.intersphinx",
  41     "sphinx.ext.napoleon",
  42     "sphinx.ext.todo",
  43     "sphinx.ext.viewcode",
  44     "nbsphinx",
  45 ]
  46 
  47 # List of patterns, relative to source directory, that match files and
  48 # directories to ignore when looking for source files.
  49 # This pattern also affects html_static_path and html_extra_path.
  50 exclude_patterns = ["_build"]
  51 
  52 # Sphinx 3.0+ required for:
  53 # autodoc_typehints = "description"
  54 needs_sphinx = "3.0"
  55 
  56 nitpicky = True
  57 nitpick_ignore = [
  58     # https://github.com/sphinx-doc/sphinx/issues/8127
  59     ("py:class", ".."),
  60     # TODO: can't figure out why this isn't found
  61     ("py:class", "LightningDataModule"),
  62     # Undocumented class
  63     ("py:class", "torchvision.models.resnet.ResNet"),
  64 ]
  65 
  66 
  67 # -- Options for HTML output -------------------------------------------------
  68 
  69 # The theme to use for HTML and HTML Help pages.  See the documentation for
  70 # a list of builtin themes.
  71 html_theme = "pytorch_sphinx_theme"
  72 html_theme_path = [pytorch_sphinx_theme.get_html_theme_path()]
  73 
  74 # Theme options are theme-specific and customize the look and feel of a theme
  75 # further.  For a list of options available for each theme, see the
  76 # documentation.
  77 html_theme_options = {
  78     "collapse_navigation": False,
  79     "display_version": True,
  80     "logo_only": True,
  81     "pytorch_project": "docs",
  82     "navigation_with_keys": True,
  83     "analytics_id": "UA-209075005-1",
  84 }
  85 
  86 html_favicon = os.path.join("..", "logo", "favicon.ico")
  87 
  88 html_static_path = ["_static"]
  89 html_css_files = ["workaround.css"]
  90 
  91 # -- Extension configuration -------------------------------------------------
  92 
  93 # sphinx.ext.autodoc
  94 autodoc_default_options = {
  95     "members": True,
  96     "special-members": True,
  97     "show-inheritance": True,
  98 }
  99 autodoc_member_order = "bysource"
  100 autodoc_typehints = "description"
  101 
  102 # sphinx.ext.intersphinx
  103 intersphinx_mapping = {
  104     "matplotlib": ("https://matplotlib.org/stable/", None),
  105     "python": ("https://docs.python.org/3", None),
  106     "pytorch-lightning": ("https://pytorch-lightning.readthedocs.io/en/latest/", None),
  107     "rasterio": ("https://rasterio.readthedocs.io/en/latest/", None),
  108     "rtree": ("https://rtree.readthedocs.io/en/latest/", None),
  109     "torch": ("https://pytorch.org/docs/stable", None),
  110     "torchvision": ("https://pytorch.org/vision/stable", None),
  111 }
  112 
  113 # nbsphinx
  114 nbsphinx_execute = "never"
  115 # TODO: branch/tag should change depending on which version of docs you look at
  116 # TODO: width option of image directive is broken, see:
  117 # https://github.com/pytorch/pytorch_sphinx_theme/issues/140
  118 nbsphinx_prolog = """
  119 {% set colab = "https://colab.research.google.com" %}
  120 {% set repo = "microsoft/torchgeo" %}
  121 {% set branch = "main" %}
  122 
  123 .. image:: {{ colab }}/assets/colab-badge.svg
  124    :class: colabbadge
  125    :alt: Open in Colab
  126    :target: {{ colab }}/github/{{ repo }}/blob/{{ branch }}/docs/{{ env.docname }}.ipynb
  127 """
  128 
  129 # Disables requirejs in nbsphinx to enable compatibility with the pytorch_sphinx_theme
  130 # See more information here https://github.com/spatialaudio/nbsphinx/issues/599
  131 # NOTE: This will likely break nbsphinx widgets
  132 nbsphinx_requirejs_path = ""
  133
  ```
  --- END FILES ---
  
  Please first localize the bug based on the issue statement, and then generate a patch according to the `git diff` format fenced by three backticks.
  
  Here is an example:
  
  ```diff
  diff --git a/examples/server_async.py b/examples/server_async.py
  --- a/examples/server_async.py
  +++ b/examples/server_async.py
  @@ -313,4 +313,4 @@
   
   
   if __name__ == "__main__":
  -    asyncio.run(run_async_server("."), debug=True)
  +    asyncio.run(run_async_server(), debug=True)
  diff --git a/examples/server_sync.py b/examples/server_sync.py
  --- a/examples/server_sync.py
  +++ b/examples/server_sync.py
  @@ -313,5 +313,5 @@
   
   
   if __name__ == "__main__":
  -    server = run_sync_server(".")
  +    server = run_sync_server()
       server.shutdown()
  
  ```
  
golden_diff: diff --git a/docs/conf.py b/docs/conf.py
  --- a/docs/conf.py
  +++ b/docs/conf.py
  @@ -118,7 +118,11 @@
   nbsphinx_prolog = """
   {% set colab = "https://colab.research.google.com" %}
   {% set repo = "microsoft/torchgeo" %}
  -{% set branch = "main" %}
  +{% if "dev" in env.config.release %}
  +    {% set branch = "main" %}
  +{% else %}
  +    {% set branch = "releases/v" ~ env.config.version %}
  +{% endif %}
   
   .. image:: {{ colab }}/assets/colab-badge.svg
      :class: colabbadge
  
verification_info: {"golden_diff": "diff --git a/docs/conf.py b/docs/conf.py\n--- a/docs/conf.py\n+++ b/docs/conf.py\n@@ -118,7 +118,11 @@\n nbsphinx_prolog = \"\"\"\n {% set colab = \"https://colab.research.google.com\" %}\n {% set repo = \"microsoft/torchgeo\" %}\n-{% set branch = \"main\" %}\n+{% if \"dev\" in env.config.release %}\n+    {% set branch = \"main\" %}\n+{% else %}\n+    {% set branch = \"releases/v\" ~ env.config.version %}\n+{% endif %}\n \n .. image:: {{ colab }}/assets/colab-badge.svg\n    :class: colabbadge\n", "issue": "Open in Colab URL broken in 0.1.1\nThe latest 0.1.1 release broke the \"Open in Colab\" URL in our tutorials. Still trying to fix this.\n", "before_files": [{"content": "# Copyright (c) Microsoft Corporation. All rights reserved.\n# Licensed under the MIT License.\n\n# Configuration file for the Sphinx documentation builder.\n#\n# This file only contains a selection of the most common options. For a full\n# list see the documentation:\n# https://www.sphinx-doc.org/en/master/usage/configuration.html\n\n# -- Path setup --------------------------------------------------------------\n\nimport os\nimport sys\n\nimport pytorch_sphinx_theme\n\n# If extensions (or modules to document with autodoc) are in another directory,\n# add these directories to sys.path here. If the directory is relative to the\n# documentation root, use os.path.abspath to make it absolute, like shown here.\nsys.path.insert(0, os.path.abspath(\"..\"))\n\nimport torchgeo  # noqa: E402\n\n# -- Project information -----------------------------------------------------\n\nproject = \"torchgeo\"\ncopyright = \"2021, Microsoft Corporation\"\nauthor = torchgeo.__author__\nversion = \".\".join(torchgeo.__version__.split(\".\")[:2])\nrelease = torchgeo.__version__\n\n\n# -- General configuration ---------------------------------------------------\n\n# Add any Sphinx extension module names here, as strings. They can be\n# extensions coming with Sphinx (named 'sphinx.ext.*') or your custom\n# ones.\nextensions = [\n    \"sphinx.ext.autodoc\",\n    \"sphinx.ext.intersphinx\",\n    \"sphinx.ext.napoleon\",\n    \"sphinx.ext.todo\",\n    \"sphinx.ext.viewcode\",\n    \"nbsphinx\",\n]\n\n# List of patterns, relative to source directory, that match files and\n# directories to ignore when looking for source files.\n# This pattern also affects html_static_path and html_extra_path.\nexclude_patterns = [\"_build\"]\n\n# Sphinx 3.0+ required for:\n# autodoc_typehints = \"description\"\nneeds_sphinx = \"3.0\"\n\nnitpicky = True\nnitpick_ignore = [\n    # https://github.com/sphinx-doc/sphinx/issues/8127\n    (\"py:class\", \"..\"),\n    # TODO: can't figure out why this isn't found\n    (\"py:class\", \"LightningDataModule\"),\n    # Undocumented class\n    (\"py:class\", \"torchvision.models.resnet.ResNet\"),\n]\n\n\n# -- Options for HTML output -------------------------------------------------\n\n# The theme to use for HTML and HTML Help pages.  See the documentation for\n# a list of builtin themes.\nhtml_theme = \"pytorch_sphinx_theme\"\nhtml_theme_path = [pytorch_sphinx_theme.get_html_theme_path()]\n\n# Theme options are theme-specific and customize the look and feel of a theme\n# further.  For a list of options available for each theme, see the\n# documentation.\nhtml_theme_options = {\n    \"collapse_navigation\": False,\n    \"display_version\": True,\n    \"logo_only\": True,\n    \"pytorch_project\": \"docs\",\n    \"navigation_with_keys\": True,\n    \"analytics_id\": \"UA-209075005-1\",\n}\n\nhtml_favicon = os.path.join(\"..\", \"logo\", \"favicon.ico\")\n\nhtml_static_path = [\"_static\"]\nhtml_css_files = [\"workaround.css\"]\n\n# -- Extension configuration -------------------------------------------------\n\n# sphinx.ext.autodoc\nautodoc_default_options = {\n    \"members\": True,\n    \"special-members\": True,\n    \"show-inheritance\": True,\n}\nautodoc_member_order = \"bysource\"\nautodoc_typehints = \"description\"\n\n# sphinx.ext.intersphinx\nintersphinx_mapping = {\n    \"matplotlib\": (\"https://matplotlib.org/stable/\", None),\n    \"python\": (\"https://docs.python.org/3\", None),\n    \"pytorch-lightning\": (\"https://pytorch-lightning.readthedocs.io/en/latest/\", None),\n    \"rasterio\": (\"https://rasterio.readthedocs.io/en/latest/\", None),\n    \"rtree\": (\"https://rtree.readthedocs.io/en/latest/\", None),\n    \"torch\": (\"https://pytorch.org/docs/stable\", None),\n    \"torchvision\": (\"https://pytorch.org/vision/stable\", None),\n}\n\n# nbsphinx\nnbsphinx_execute = \"never\"\n# TODO: branch/tag should change depending on which version of docs you look at\n# TODO: width option of image directive is broken, see:\n# https://github.com/pytorch/pytorch_sphinx_theme/issues/140\nnbsphinx_prolog = \"\"\"\n{% set colab = \"https://colab.research.google.com\" %}\n{% set repo = \"microsoft/torchgeo\" %}\n{% set branch = \"main\" %}\n\n.. image:: {{ colab }}/assets/colab-badge.svg\n   :class: colabbadge\n   :alt: Open in Colab\n   :target: {{ colab }}/github/{{ repo }}/blob/{{ branch }}/docs/{{ env.docname }}.ipynb\n\"\"\"\n\n# Disables requirejs in nbsphinx to enable compatibility with the pytorch_sphinx_theme\n# See more information here https://github.com/spatialaudio/nbsphinx/issues/599\n# NOTE: This will likely break nbsphinx widgets\nnbsphinx_requirejs_path = \"\"\n", "path": "docs/conf.py"}], "after_files": [{"content": "# Copyright (c) Microsoft Corporation. All rights reserved.\n# Licensed under the MIT License.\n\n# Configuration file for the Sphinx documentation builder.\n#\n# This file only contains a selection of the most common options. For a full\n# list see the documentation:\n# https://www.sphinx-doc.org/en/master/usage/configuration.html\n\n# -- Path setup --------------------------------------------------------------\n\nimport os\nimport sys\n\nimport pytorch_sphinx_theme\n\n# If extensions (or modules to document with autodoc) are in another directory,\n# add these directories to sys.path here. If the directory is relative to the\n# documentation root, use os.path.abspath to make it absolute, like shown here.\nsys.path.insert(0, os.path.abspath(\"..\"))\n\nimport torchgeo  # noqa: E402\n\n# -- Project information -----------------------------------------------------\n\nproject = \"torchgeo\"\ncopyright = \"2021, Microsoft Corporation\"\nauthor = torchgeo.__author__\nversion = \".\".join(torchgeo.__version__.split(\".\")[:2])\nrelease = torchgeo.__version__\n\n\n# -- General configuration ---------------------------------------------------\n\n# Add any Sphinx extension module names here, as strings. They can be\n# extensions coming with Sphinx (named 'sphinx.ext.*') or your custom\n# ones.\nextensions = [\n    \"sphinx.ext.autodoc\",\n    \"sphinx.ext.intersphinx\",\n    \"sphinx.ext.napoleon\",\n    \"sphinx.ext.todo\",\n    \"sphinx.ext.viewcode\",\n    \"nbsphinx\",\n]\n\n# List of patterns, relative to source directory, that match files and\n# directories to ignore when looking for source files.\n# This pattern also affects html_static_path and html_extra_path.\nexclude_patterns = [\"_build\"]\n\n# Sphinx 3.0+ required for:\n# autodoc_typehints = \"description\"\nneeds_sphinx = \"3.0\"\n\nnitpicky = True\nnitpick_ignore = [\n    # https://github.com/sphinx-doc/sphinx/issues/8127\n    (\"py:class\", \"..\"),\n    # TODO: can't figure out why this isn't found\n    (\"py:class\", \"LightningDataModule\"),\n    # Undocumented class\n    (\"py:class\", \"torchvision.models.resnet.ResNet\"),\n]\n\n\n# -- Options for HTML output -------------------------------------------------\n\n# The theme to use for HTML and HTML Help pages.  See the documentation for\n# a list of builtin themes.\nhtml_theme = \"pytorch_sphinx_theme\"\nhtml_theme_path = [pytorch_sphinx_theme.get_html_theme_path()]\n\n# Theme options are theme-specific and customize the look and feel of a theme\n# further.  For a list of options available for each theme, see the\n# documentation.\nhtml_theme_options = {\n    \"collapse_navigation\": False,\n    \"display_version\": True,\n    \"logo_only\": True,\n    \"pytorch_project\": \"docs\",\n    \"navigation_with_keys\": True,\n    \"analytics_id\": \"UA-209075005-1\",\n}\n\nhtml_favicon = os.path.join(\"..\", \"logo\", \"favicon.ico\")\n\nhtml_static_path = [\"_static\"]\nhtml_css_files = [\"workaround.css\"]\n\n# -- Extension configuration -------------------------------------------------\n\n# sphinx.ext.autodoc\nautodoc_default_options = {\n    \"members\": True,\n    \"special-members\": True,\n    \"show-inheritance\": True,\n}\nautodoc_member_order = \"bysource\"\nautodoc_typehints = \"description\"\n\n# sphinx.ext.intersphinx\nintersphinx_mapping = {\n    \"matplotlib\": (\"https://matplotlib.org/stable/\", None),\n    \"python\": (\"https://docs.python.org/3\", None),\n    \"pytorch-lightning\": (\"https://pytorch-lightning.readthedocs.io/en/latest/\", None),\n    \"rasterio\": (\"https://rasterio.readthedocs.io/en/latest/\", None),\n    \"rtree\": (\"https://rtree.readthedocs.io/en/latest/\", None),\n    \"torch\": (\"https://pytorch.org/docs/stable\", None),\n    \"torchvision\": (\"https://pytorch.org/vision/stable\", None),\n}\n\n# nbsphinx\nnbsphinx_execute = \"never\"\n# TODO: branch/tag should change depending on which version of docs you look at\n# TODO: width option of image directive is broken, see:\n# https://github.com/pytorch/pytorch_sphinx_theme/issues/140\nnbsphinx_prolog = \"\"\"\n{% set colab = \"https://colab.research.google.com\" %}\n{% set repo = \"microsoft/torchgeo\" %}\n{% if \"dev\" in env.config.release %}\n    {% set branch = \"main\" %}\n{% else %}\n    {% set branch = \"releases/v\" ~ env.config.version %}\n{% endif %}\n\n.. image:: {{ colab }}/assets/colab-badge.svg\n   :class: colabbadge\n   :alt: Open in Colab\n   :target: {{ colab }}/github/{{ repo }}/blob/{{ branch }}/docs/{{ env.docname }}.ipynb\n\"\"\"\n\n# Disables requirejs in nbsphinx to enable compatibility with the pytorch_sphinx_theme\n# See more information here https://github.com/spatialaudio/nbsphinx/issues/599\n# NOTE: This will likely break nbsphinx widgets\nnbsphinx_requirejs_path = \"\"\n", "path": "docs/conf.py"}]}