We are currently solving the following issue within our repository. Here is the issue text:
--- BEGIN ISSUE ---
[Ax-Plugin] Add support for Python 3.9
Python 3.9 support pending on scikit 2.4.0 release. Relevant comment: https://github.com/scikit-learn/scikit-learn/issues/18621#issuecomment-733078676

Related to #1062 

--- END ISSUE ---

Below are some code segments, each from a relevant file. One or more of these files may contain bugs.
--- BEGIN FILES ---
Path: `plugins/hydra_ax_sweeper/setup.py`
Content:
```
1 # Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
2 # type: ignore
3 from setuptools import find_namespace_packages, setup
4 
5 with open("README.md", "r") as fh:
6     LONG_DESC = fh.read()
7     setup(
8         name="hydra-ax-sweeper",
9         version="1.1.0rc1",
10         author="Omry Yadan, Shagun Sodhani",
11         author_email="omry@fb.com, sshagunsodhani@gmail.com",
12         description="Hydra Ax Sweeper plugin",
13         long_description=LONG_DESC,
14         long_description_content_type="text/markdown",
15         url="https://github.com/facebookresearch/hydra/",
16         packages=find_namespace_packages(include=["hydra_plugins.*"]),
17         classifiers=[
18             "License :: OSI Approved :: MIT License",
19             "Programming Language :: Python :: 3.7",
20             "Programming Language :: Python :: 3.8",
21             # "Programming Language :: Python :: 3.9",
22             "Operating System :: POSIX :: Linux",
23             "Operating System :: MacOS",
24             "Development Status :: 4 - Beta",
25         ],
26         install_requires=["hydra-core>=1.0.0", "ax-platform>=0.1.13"],
27         include_package_data=True,
28     )
29
```
--- 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/plugins/hydra_ax_sweeper/setup.py b/plugins/hydra_ax_sweeper/setup.py
--- a/plugins/hydra_ax_sweeper/setup.py
+++ b/plugins/hydra_ax_sweeper/setup.py
@@ -18,7 +18,7 @@
             "License :: OSI Approved :: MIT License",
             "Programming Language :: Python :: 3.7",
             "Programming Language :: Python :: 3.8",
-            # "Programming Language :: Python :: 3.9",
+            "Programming Language :: Python :: 3.9",
             "Operating System :: POSIX :: Linux",
             "Operating System :: MacOS",
             "Development Status :: 4 - Beta",

{"golden_diff": "diff --git a/plugins/hydra_ax_sweeper/setup.py b/plugins/hydra_ax_sweeper/setup.py\n--- a/plugins/hydra_ax_sweeper/setup.py\n+++ b/plugins/hydra_ax_sweeper/setup.py\n@@ -18,7 +18,7 @@\n             \"License :: OSI Approved :: MIT License\",\n             \"Programming Language :: Python :: 3.7\",\n             \"Programming Language :: Python :: 3.8\",\n-            # \"Programming Language :: Python :: 3.9\",\n+            \"Programming Language :: Python :: 3.9\",\n             \"Operating System :: POSIX :: Linux\",\n             \"Operating System :: MacOS\",\n             \"Development Status :: 4 - Beta\",\n", "issue": "[Ax-Plugin] Add support for Python 3.9\nPython 3.9 support pending on scikit 2.4.0 release. Relevant comment: https://github.com/scikit-learn/scikit-learn/issues/18621#issuecomment-733078676\r\n\r\nRelated to #1062 \n", "before_files": [{"content": "# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved\n# type: ignore\nfrom setuptools import find_namespace_packages, setup\n\nwith open(\"README.md\", \"r\") as fh:\n    LONG_DESC = fh.read()\n    setup(\n        name=\"hydra-ax-sweeper\",\n        version=\"1.1.0rc1\",\n        author=\"Omry Yadan, Shagun Sodhani\",\n        author_email=\"omry@fb.com, sshagunsodhani@gmail.com\",\n        description=\"Hydra Ax Sweeper plugin\",\n        long_description=LONG_DESC,\n        long_description_content_type=\"text/markdown\",\n        url=\"https://github.com/facebookresearch/hydra/\",\n        packages=find_namespace_packages(include=[\"hydra_plugins.*\"]),\n        classifiers=[\n            \"License :: OSI Approved :: MIT License\",\n            \"Programming Language :: Python :: 3.7\",\n            \"Programming Language :: Python :: 3.8\",\n            # \"Programming Language :: Python :: 3.9\",\n            \"Operating System :: POSIX :: Linux\",\n            \"Operating System :: MacOS\",\n            \"Development Status :: 4 - Beta\",\n        ],\n        install_requires=[\"hydra-core>=1.0.0\", \"ax-platform>=0.1.13\"],\n        include_package_data=True,\n    )\n", "path": "plugins/hydra_ax_sweeper/setup.py"}], "after_files": [{"content": "# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved\n# type: ignore\nfrom setuptools import find_namespace_packages, setup\n\nwith open(\"README.md\", \"r\") as fh:\n    LONG_DESC = fh.read()\n    setup(\n        name=\"hydra-ax-sweeper\",\n        version=\"1.1.0rc1\",\n        author=\"Omry Yadan, Shagun Sodhani\",\n        author_email=\"omry@fb.com, sshagunsodhani@gmail.com\",\n        description=\"Hydra Ax Sweeper plugin\",\n        long_description=LONG_DESC,\n        long_description_content_type=\"text/markdown\",\n        url=\"https://github.com/facebookresearch/hydra/\",\n        packages=find_namespace_packages(include=[\"hydra_plugins.*\"]),\n        classifiers=[\n            \"License :: OSI Approved :: MIT License\",\n            \"Programming Language :: Python :: 3.7\",\n            \"Programming Language :: Python :: 3.8\",\n            \"Programming Language :: Python :: 3.9\",\n            \"Operating System :: POSIX :: Linux\",\n            \"Operating System :: MacOS\",\n            \"Development Status :: 4 - Beta\",\n        ],\n        install_requires=[\"hydra-core>=1.0.0\", \"ax-platform>=0.1.13\"],\n        include_package_data=True,\n    )\n", "path": "plugins/hydra_ax_sweeper/setup.py"}]}