We are currently solving the following issue within our repository. Here is the issue text:
--- BEGIN ISSUE ---
Adding pytest in an MPI environment
There are some functions that run in a distributed environment that have not been fully verified, so some pytests will be run in MPI to verify them.

https://github.com/pfnet/pytorch-pfn-extras/blob/266e8bde2c2a1a6aa3f8648d49e758975c8b436a/tests/pytorch_pfn_extras_tests/training_tests/extensions_tests/test_distributed_snapshot.py#L40-L46


--- 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 import os
2 
3 import setuptools
4 
5 here = os.path.abspath(os.path.dirname(__file__))
6 # Get __version__ variable
7 exec(open(os.path.join(here, "pytorch_pfn_extras", "_version.py")).read())
8 
9 
10 setuptools.setup(
11     name="pytorch-pfn-extras",
12     version=__version__,  # NOQA
13     description="Supplementary components to accelerate research and "
14     "development in PyTorch.",
15     author="Preferred Networks, Inc.",
16     license="MIT License",
17     install_requires=["numpy", "packaging", "torch", "typing-extensions>=3.10"],
18     extras_require={
19         "test": [
20             "pytest",
21             "onnxruntime",
22             "torchvision",
23             "torchaudio",
24             "pysen",
25             "black==23.3.0",
26             "flake8==4.0.1",
27             "isort==5.10.1",
28             "mypy==1.3.0",
29             "types-PyYAML",
30             "types-setuptools",
31             "matplotlib",
32             "tensorboard",
33             "ipython",
34             "ipywidgets",
35             "pandas",
36             "optuna",
37             "onnx",
38             "pytorch-ignite",
39         ],
40         "onnx": ["onnx"],
41     },
42     python_requires=">=3.6.0",
43     packages=setuptools.find_packages(exclude=["tests", "tests.*"]),
44     package_data={"pytorch_pfn_extras": ["py.typed"]},
45 )
46
```
--- 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
@@ -16,27 +16,6 @@
     license="MIT License",
     install_requires=["numpy", "packaging", "torch", "typing-extensions>=3.10"],
     extras_require={
-        "test": [
-            "pytest",
-            "onnxruntime",
-            "torchvision",
-            "torchaudio",
-            "pysen",
-            "black==23.3.0",
-            "flake8==4.0.1",
-            "isort==5.10.1",
-            "mypy==1.3.0",
-            "types-PyYAML",
-            "types-setuptools",
-            "matplotlib",
-            "tensorboard",
-            "ipython",
-            "ipywidgets",
-            "pandas",
-            "optuna",
-            "onnx",
-            "pytorch-ignite",
-        ],
         "onnx": ["onnx"],
     },
     python_requires=">=3.6.0",

{"golden_diff": "diff --git a/setup.py b/setup.py\n--- a/setup.py\n+++ b/setup.py\n@@ -16,27 +16,6 @@\n     license=\"MIT License\",\n     install_requires=[\"numpy\", \"packaging\", \"torch\", \"typing-extensions>=3.10\"],\n     extras_require={\n-        \"test\": [\n-            \"pytest\",\n-            \"onnxruntime\",\n-            \"torchvision\",\n-            \"torchaudio\",\n-            \"pysen\",\n-            \"black==23.3.0\",\n-            \"flake8==4.0.1\",\n-            \"isort==5.10.1\",\n-            \"mypy==1.3.0\",\n-            \"types-PyYAML\",\n-            \"types-setuptools\",\n-            \"matplotlib\",\n-            \"tensorboard\",\n-            \"ipython\",\n-            \"ipywidgets\",\n-            \"pandas\",\n-            \"optuna\",\n-            \"onnx\",\n-            \"pytorch-ignite\",\n-        ],\n         \"onnx\": [\"onnx\"],\n     },\n     python_requires=\">=3.6.0\",\n", "issue": "Adding pytest in an MPI environment\nThere are some functions that run in a distributed environment that have not been fully verified, so some pytests will be run in MPI to verify them.\r\n\r\nhttps://github.com/pfnet/pytorch-pfn-extras/blob/266e8bde2c2a1a6aa3f8648d49e758975c8b436a/tests/pytorch_pfn_extras_tests/training_tests/extensions_tests/test_distributed_snapshot.py#L40-L46\r\n\n", "before_files": [{"content": "import os\n\nimport setuptools\n\nhere = os.path.abspath(os.path.dirname(__file__))\n# Get __version__ variable\nexec(open(os.path.join(here, \"pytorch_pfn_extras\", \"_version.py\")).read())\n\n\nsetuptools.setup(\n    name=\"pytorch-pfn-extras\",\n    version=__version__,  # NOQA\n    description=\"Supplementary components to accelerate research and \"\n    \"development in PyTorch.\",\n    author=\"Preferred Networks, Inc.\",\n    license=\"MIT License\",\n    install_requires=[\"numpy\", \"packaging\", \"torch\", \"typing-extensions>=3.10\"],\n    extras_require={\n        \"test\": [\n            \"pytest\",\n            \"onnxruntime\",\n            \"torchvision\",\n            \"torchaudio\",\n            \"pysen\",\n            \"black==23.3.0\",\n            \"flake8==4.0.1\",\n            \"isort==5.10.1\",\n            \"mypy==1.3.0\",\n            \"types-PyYAML\",\n            \"types-setuptools\",\n            \"matplotlib\",\n            \"tensorboard\",\n            \"ipython\",\n            \"ipywidgets\",\n            \"pandas\",\n            \"optuna\",\n            \"onnx\",\n            \"pytorch-ignite\",\n        ],\n        \"onnx\": [\"onnx\"],\n    },\n    python_requires=\">=3.6.0\",\n    packages=setuptools.find_packages(exclude=[\"tests\", \"tests.*\"]),\n    package_data={\"pytorch_pfn_extras\": [\"py.typed\"]},\n)\n", "path": "setup.py"}], "after_files": [{"content": "import os\n\nimport setuptools\n\nhere = os.path.abspath(os.path.dirname(__file__))\n# Get __version__ variable\nexec(open(os.path.join(here, \"pytorch_pfn_extras\", \"_version.py\")).read())\n\n\nsetuptools.setup(\n    name=\"pytorch-pfn-extras\",\n    version=__version__,  # NOQA\n    description=\"Supplementary components to accelerate research and \"\n    \"development in PyTorch.\",\n    author=\"Preferred Networks, Inc.\",\n    license=\"MIT License\",\n    install_requires=[\"numpy\", \"packaging\", \"torch\", \"typing-extensions>=3.10\"],\n    extras_require={\n        \"onnx\": [\"onnx\"],\n    },\n    python_requires=\">=3.6.0\",\n    packages=setuptools.find_packages(exclude=[\"tests\", \"tests.*\"]),\n    package_data={\"pytorch_pfn_extras\": [\"py.typed\"]},\n)\n", "path": "setup.py"}]}