prompt: 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()
  
  ```
  
golden_diff: 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']
  
verification_info: {"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"}]}