We are currently solving the following issue within our repository. Here is the issue text:
--- BEGIN ISSUE ---
Ensure type information is packaged with installable
**Describe the feature you'd like**
As a user, I want to be able to type check my code that calls into trieste. It's not clear at the moment if this is currently possible, given there is no py.typed file.

See the python PEP for more info https://www.python.org/dev/peps/pep-0561/

--- 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 2020 The Trieste Contributors
2 #
3 # Licensed under the Apache License, Version 2.0 (the "License");
4 # you may not use this file except in compliance with the License.
5 # You may obtain a copy of the License at
6 #
7 #     http://www.apache.org/licenses/LICENSE-2.0
8 #
9 # Unless required by applicable law or agreed to in writing, software
10 # distributed under the License is distributed on an "AS IS" BASIS,
11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 # See the License for the specific language governing permissions and
13 # limitations under the License.
14 
15 from setuptools import find_packages, setup
16 
17 with open("README.md", "r") as file:
18     long_description = file.read()
19 
20 setup(
21     name="trieste",
22     version="0.3.1",
23     author="The Trieste contributors",
24     author_email="labs@secondmind.ai",
25     description="A Bayesian optimization research toolbox built on TensorFlow",
26     long_description=long_description,
27     long_description_content_type="text/markdown",
28     url="https://github.com/secondmind-labs/trieste",
29     packages=find_packages(include=("trieste*",)),
30     classifiers=[
31         "Programming Language :: Python :: 3.7",
32         "License :: OSI Approved :: Apache Software License",
33         "Operating System :: OS Independent",
34     ],
35     python_requires="~=3.7",
36     install_requires=[
37         "absl-py",
38         "gpflow==2.1.*",
39         "numpy",
40         # tensorflow!=2.2.0,!=2.3.0 because of https://github.com/advisories/GHSA-8fxw-76px-3rxv
41         "tensorflow>=2.1,!=2.2.0,!=2.3.0",
42         "tensorflow-probability>=0.9",
43     ],
44 )
45
```
--- 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
@@ -27,6 +27,9 @@
     long_description_content_type="text/markdown",
     url="https://github.com/secondmind-labs/trieste",
     packages=find_packages(include=("trieste*",)),
+    package_data={
+        "trieste": ["py.typed"],
+    },
     classifiers=[
         "Programming Language :: Python :: 3.7",
         "License :: OSI Approved :: Apache Software License",

{"golden_diff": "diff --git a/setup.py b/setup.py\n--- a/setup.py\n+++ b/setup.py\n@@ -27,6 +27,9 @@\n     long_description_content_type=\"text/markdown\",\n     url=\"https://github.com/secondmind-labs/trieste\",\n     packages=find_packages(include=(\"trieste*\",)),\n+    package_data={\n+        \"trieste\": [\"py.typed\"],\n+    },\n     classifiers=[\n         \"Programming Language :: Python :: 3.7\",\n         \"License :: OSI Approved :: Apache Software License\",\n", "issue": "Ensure type information is packaged with installable\n**Describe the feature you'd like**\r\nAs a user, I want to be able to type check my code that calls into trieste. It's not clear at the moment if this is currently possible, given there is no py.typed file.\r\n\r\nSee the python PEP for more info https://www.python.org/dev/peps/pep-0561/\n", "before_files": [{"content": "# Copyright 2020 The Trieste Contributors\n#\n# Licensed under the Apache License, Version 2.0 (the \"License\");\n# you may not use this file except in compliance with the License.\n# You may obtain a copy of the License at\n#\n#     http://www.apache.org/licenses/LICENSE-2.0\n#\n# Unless required by applicable law or agreed to in writing, software\n# distributed under the License is distributed on an \"AS IS\" BASIS,\n# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n# See the License for the specific language governing permissions and\n# limitations under the License.\n\nfrom setuptools import find_packages, setup\n\nwith open(\"README.md\", \"r\") as file:\n    long_description = file.read()\n\nsetup(\n    name=\"trieste\",\n    version=\"0.3.1\",\n    author=\"The Trieste contributors\",\n    author_email=\"labs@secondmind.ai\",\n    description=\"A Bayesian optimization research toolbox built on TensorFlow\",\n    long_description=long_description,\n    long_description_content_type=\"text/markdown\",\n    url=\"https://github.com/secondmind-labs/trieste\",\n    packages=find_packages(include=(\"trieste*\",)),\n    classifiers=[\n        \"Programming Language :: Python :: 3.7\",\n        \"License :: OSI Approved :: Apache Software License\",\n        \"Operating System :: OS Independent\",\n    ],\n    python_requires=\"~=3.7\",\n    install_requires=[\n        \"absl-py\",\n        \"gpflow==2.1.*\",\n        \"numpy\",\n        # tensorflow!=2.2.0,!=2.3.0 because of https://github.com/advisories/GHSA-8fxw-76px-3rxv\n        \"tensorflow>=2.1,!=2.2.0,!=2.3.0\",\n        \"tensorflow-probability>=0.9\",\n    ],\n)\n", "path": "setup.py"}], "after_files": [{"content": "# Copyright 2020 The Trieste Contributors\n#\n# Licensed under the Apache License, Version 2.0 (the \"License\");\n# you may not use this file except in compliance with the License.\n# You may obtain a copy of the License at\n#\n#     http://www.apache.org/licenses/LICENSE-2.0\n#\n# Unless required by applicable law or agreed to in writing, software\n# distributed under the License is distributed on an \"AS IS\" BASIS,\n# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n# See the License for the specific language governing permissions and\n# limitations under the License.\n\nfrom setuptools import find_packages, setup\n\nwith open(\"README.md\", \"r\") as file:\n    long_description = file.read()\n\nsetup(\n    name=\"trieste\",\n    version=\"0.3.1\",\n    author=\"The Trieste contributors\",\n    author_email=\"labs@secondmind.ai\",\n    description=\"A Bayesian optimization research toolbox built on TensorFlow\",\n    long_description=long_description,\n    long_description_content_type=\"text/markdown\",\n    url=\"https://github.com/secondmind-labs/trieste\",\n    packages=find_packages(include=(\"trieste*\",)),\n    package_data={\n        \"trieste\": [\"py.typed\"],\n    },\n    classifiers=[\n        \"Programming Language :: Python :: 3.7\",\n        \"License :: OSI Approved :: Apache Software License\",\n        \"Operating System :: OS Independent\",\n    ],\n    python_requires=\"~=3.7\",\n    install_requires=[\n        \"absl-py\",\n        \"gpflow==2.1.*\",\n        \"numpy\",\n        # tensorflow!=2.2.0,!=2.3.0 because of https://github.com/advisories/GHSA-8fxw-76px-3rxv\n        \"tensorflow>=2.1,!=2.2.0,!=2.3.0\",\n        \"tensorflow-probability>=0.9\",\n    ],\n)\n", "path": "setup.py"}]}