We are currently solving the following issue within our repository. Here is the issue text:
--- BEGIN ISSUE ---
Add Colab Example
* Add Example Jupyter notebook to the examples folder
* Add "Open in Colab" to the README.md


--- END ISSUE ---

Below are some code segments, each from a relevant file. One or more of these files may contain bugs.
--- BEGIN FILES ---
Path: `setup.py`
Content:
```
1 # Copyright 2021 MosaicML. All Rights Reserved.
2 
3 import os
4 import sys
5 
6 import setuptools
7 from setuptools import setup
8 
9 
10 def package_files(directory):
11     # from https://stackoverflow.com/a/36693250
12     paths = []
13     for (path, directories, filenames) in os.walk(directory):
14         for filename in filenames:
15             paths.append(os.path.join('..', path, filename))
16     return paths
17 
18 
19 with open("README.md", "r", encoding="utf-8") as fh:
20     long_description = fh.read()
21 
22 install_requires = [
23     "pyyaml>=5.4.1",
24     "tqdm>=4.62.3",
25     "torchmetrics>=0.5.1",
26     "torch_optimizer==0.1.0",
27     "torchvision>=0.9.0",
28     "torch>=1.9",
29     "argparse>=1.4.0",
30     "yahp>=0.0.10",
31 ]
32 extra_deps = {}
33 
34 extra_deps['base'] = []
35 
36 extra_deps['dev'] = [
37     'junitparser>=2.1.1',
38     'coverage[toml]>=6.1.1',
39     'pytest>=6.2.0',
40     'yapf>=0.13.0',
41     'isort>=5.9.3',
42     'yamllint>=1.26.2',
43     'pytest-timeout>=1.4.2',
44     'recommonmark>=0.7.1',
45     'sphinx>=4.2.0',
46     'sphinx_copybutton>=0.4.0',
47     'sphinx_markdown_tables>=0.0.15',
48     'sphinx-argparse>=0.3.1',
49     'sphinxcontrib.katex>=0.8.6',
50     'sphinxext.opengraph>=0.4.2',
51     'sphinx_rtd_theme>=1.0.0',
52     'myst-parser>=0.15.2',
53 ]
54 extra_deps['wandb'] = ['wandb>=0.12.2']
55 
56 extra_deps['nlp'] = [
57     'transformers>=4.11.3',
58     'datasets>=1.14.0',
59 ]
60 
61 extra_deps['unet'] = [
62     'monai>=0.7.0',
63     'scikit-learn>=1.0.1',
64 ]
65 
66 extra_deps['all'] = set(dep for deps in extra_deps.values() for dep in deps)
67 
68 setup(
69     name="mosaicml",
70     version="0.2.4",
71     author="MosaicML",
72     author_email="team@mosaicml.com",
73     description="composing methods for ML training efficiency",
74     long_description=long_description,
75     long_description_content_type="text/markdown",
76     url="https://github.com/mosaicml/composer",
77     include_package_data=True,
78     package_data={
79         "composer": ['py.typed'],
80         "": package_files('composer/yamls'),
81     },
82     packages=setuptools.find_packages(include=["composer"]),
83     classifiers=[
84         "Programming Language :: Python :: 3",
85     ],
86     install_requires=install_requires,
87     entry_points={
88         'console_scripts': ['composer = composer.cli.launcher:main',],
89     },
90     extras_require=extra_deps,
91     dependency_links=['https://developer.download.nvidia.com/compute/redist'],
92     python_requires='>=3.7',
93     ext_package="composer",
94 )
95 
96 # only visible if user installs with verbose -v flag
97 # Printing to stdout as not to interfere with setup.py CLI flags (e.g. --version)
98 print("*" * 20, file=sys.stderr)
99 print(
100     "\nNOTE: For best performance, we recommend installing Pillow-SIMD "
101     "\nfor accelerated image processing operations. To install:"
102     "\n\n\t pip uninstall pillow && pip install pillow-simd\n",
103     file=sys.stderr)
104 print("*" * 20, file=sys.stderr)
105
```
--- 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()

```

diff --git a/setup.py b/setup.py
--- a/setup.py
+++ b/setup.py
@@ -49,6 +49,7 @@
     'sphinxcontrib.katex>=0.8.6',
     'sphinxext.opengraph>=0.4.2',
     'sphinx_rtd_theme>=1.0.0',
+    'testbook>=0.4.2',
     'myst-parser>=0.15.2',
 ]
 extra_deps['wandb'] = ['wandb>=0.12.2']

{"golden_diff": "diff --git a/setup.py b/setup.py\n--- a/setup.py\n+++ b/setup.py\n@@ -49,6 +49,7 @@\n     'sphinxcontrib.katex>=0.8.6',\n     'sphinxext.opengraph>=0.4.2',\n     'sphinx_rtd_theme>=1.0.0',\n+    'testbook>=0.4.2',\n     'myst-parser>=0.15.2',\n ]\n extra_deps['wandb'] = ['wandb>=0.12.2']\n", "issue": "Add Colab Example\n* Add Example Jupyter notebook to the examples folder\r\n* Add \"Open in Colab\" to the README.md\r\n\n", "before_files": [{"content": "# Copyright 2021 MosaicML. All Rights Reserved.\n\nimport os\nimport sys\n\nimport setuptools\nfrom setuptools import setup\n\n\ndef package_files(directory):\n    # from https://stackoverflow.com/a/36693250\n    paths = []\n    for (path, directories, filenames) in os.walk(directory):\n        for filename in filenames:\n            paths.append(os.path.join('..', path, filename))\n    return paths\n\n\nwith open(\"README.md\", \"r\", encoding=\"utf-8\") as fh:\n    long_description = fh.read()\n\ninstall_requires = [\n    \"pyyaml>=5.4.1\",\n    \"tqdm>=4.62.3\",\n    \"torchmetrics>=0.5.1\",\n    \"torch_optimizer==0.1.0\",\n    \"torchvision>=0.9.0\",\n    \"torch>=1.9\",\n    \"argparse>=1.4.0\",\n    \"yahp>=0.0.10\",\n]\nextra_deps = {}\n\nextra_deps['base'] = []\n\nextra_deps['dev'] = [\n    'junitparser>=2.1.1',\n    'coverage[toml]>=6.1.1',\n    'pytest>=6.2.0',\n    'yapf>=0.13.0',\n    'isort>=5.9.3',\n    'yamllint>=1.26.2',\n    'pytest-timeout>=1.4.2',\n    'recommonmark>=0.7.1',\n    'sphinx>=4.2.0',\n    'sphinx_copybutton>=0.4.0',\n    'sphinx_markdown_tables>=0.0.15',\n    'sphinx-argparse>=0.3.1',\n    'sphinxcontrib.katex>=0.8.6',\n    'sphinxext.opengraph>=0.4.2',\n    'sphinx_rtd_theme>=1.0.0',\n    'myst-parser>=0.15.2',\n]\nextra_deps['wandb'] = ['wandb>=0.12.2']\n\nextra_deps['nlp'] = [\n    'transformers>=4.11.3',\n    'datasets>=1.14.0',\n]\n\nextra_deps['unet'] = [\n    'monai>=0.7.0',\n    'scikit-learn>=1.0.1',\n]\n\nextra_deps['all'] = set(dep for deps in extra_deps.values() for dep in deps)\n\nsetup(\n    name=\"mosaicml\",\n    version=\"0.2.4\",\n    author=\"MosaicML\",\n    author_email=\"team@mosaicml.com\",\n    description=\"composing methods for ML training efficiency\",\n    long_description=long_description,\n    long_description_content_type=\"text/markdown\",\n    url=\"https://github.com/mosaicml/composer\",\n    include_package_data=True,\n    package_data={\n        \"composer\": ['py.typed'],\n        \"\": package_files('composer/yamls'),\n    },\n    packages=setuptools.find_packages(include=[\"composer\"]),\n    classifiers=[\n        \"Programming Language :: Python :: 3\",\n    ],\n    install_requires=install_requires,\n    entry_points={\n        'console_scripts': ['composer = composer.cli.launcher:main',],\n    },\n    extras_require=extra_deps,\n    dependency_links=['https://developer.download.nvidia.com/compute/redist'],\n    python_requires='>=3.7',\n    ext_package=\"composer\",\n)\n\n# only visible if user installs with verbose -v flag\n# Printing to stdout as not to interfere with setup.py CLI flags (e.g. --version)\nprint(\"*\" * 20, file=sys.stderr)\nprint(\n    \"\\nNOTE: For best performance, we recommend installing Pillow-SIMD \"\n    \"\\nfor accelerated image processing operations. To install:\"\n    \"\\n\\n\\t pip uninstall pillow && pip install pillow-simd\\n\",\n    file=sys.stderr)\nprint(\"*\" * 20, file=sys.stderr)\n", "path": "setup.py"}], "after_files": [{"content": "# Copyright 2021 MosaicML. All Rights Reserved.\n\nimport os\nimport sys\n\nimport setuptools\nfrom setuptools import setup\n\n\ndef package_files(directory):\n    # from https://stackoverflow.com/a/36693250\n    paths = []\n    for (path, directories, filenames) in os.walk(directory):\n        for filename in filenames:\n            paths.append(os.path.join('..', path, filename))\n    return paths\n\n\nwith open(\"README.md\", \"r\", encoding=\"utf-8\") as fh:\n    long_description = fh.read()\n\ninstall_requires = [\n    \"pyyaml>=5.4.1\",\n    \"tqdm>=4.62.3\",\n    \"torchmetrics>=0.5.1\",\n    \"torch_optimizer==0.1.0\",\n    \"torchvision>=0.9.0\",\n    \"torch>=1.9\",\n    \"argparse>=1.4.0\",\n    \"yahp>=0.0.10\",\n]\nextra_deps = {}\n\nextra_deps['base'] = []\n\nextra_deps['dev'] = [\n    'junitparser>=2.1.1',\n    'coverage[toml]>=6.1.1',\n    'pytest>=6.2.0',\n    'yapf>=0.13.0',\n    'isort>=5.9.3',\n    'yamllint>=1.26.2',\n    'pytest-timeout>=1.4.2',\n    'recommonmark>=0.7.1',\n    'sphinx>=4.2.0',\n    'sphinx_copybutton>=0.4.0',\n    'sphinx_markdown_tables>=0.0.15',\n    'sphinx-argparse>=0.3.1',\n    'sphinxcontrib.katex>=0.8.6',\n    'sphinxext.opengraph>=0.4.2',\n    'sphinx_rtd_theme>=1.0.0',\n    'testbook>=0.4.2',\n    'myst-parser>=0.15.2',\n]\nextra_deps['wandb'] = ['wandb>=0.12.2']\n\nextra_deps['nlp'] = [\n    'transformers>=4.11.3',\n    'datasets>=1.14.0',\n]\n\nextra_deps['unet'] = [\n    'monai>=0.7.0',\n    'scikit-learn>=1.0.1',\n]\n\nextra_deps['all'] = set(dep for deps in extra_deps.values() for dep in deps)\n\nsetup(\n    name=\"mosaicml\",\n    version=\"0.2.4\",\n    author=\"MosaicML\",\n    author_email=\"team@mosaicml.com\",\n    description=\"composing methods for ML training efficiency\",\n    long_description=long_description,\n    long_description_content_type=\"text/markdown\",\n    url=\"https://github.com/mosaicml/composer\",\n    include_package_data=True,\n    package_data={\n        \"composer\": ['py.typed'],\n        \"\": package_files('composer/yamls'),\n    },\n    packages=setuptools.find_packages(include=[\"composer\"]),\n    classifiers=[\n        \"Programming Language :: Python :: 3\",\n    ],\n    install_requires=install_requires,\n    entry_points={\n        'console_scripts': ['composer = composer.cli.launcher:main',],\n    },\n    extras_require=extra_deps,\n    dependency_links=['https://developer.download.nvidia.com/compute/redist'],\n    python_requires='>=3.7',\n    ext_package=\"composer\",\n)\n\n# only visible if user installs with verbose -v flag\n# Printing to stdout as not to interfere with setup.py CLI flags (e.g. --version)\nprint(\"*\" * 20, file=sys.stderr)\nprint(\n    \"\\nNOTE: For best performance, we recommend installing Pillow-SIMD \"\n    \"\\nfor accelerated image processing operations. To install:\"\n    \"\\n\\n\\t pip uninstall pillow && pip install pillow-simd\\n\",\n    file=sys.stderr)\nprint(\"*\" * 20, file=sys.stderr)\n", "path": "setup.py"}]}