From 79a30912b8d6e9867e56a579cf93d56bf30c955e Mon Sep 17 00:00:00 2001
From: Jared Roesch <roeschinc@gmail.com>
Date: Mon, 30 Oct 2023 14:50:47 -0700
Subject: [PATCH] Add py.typed so consumers of vLLM can get type checking
 (#1509)

* Add py.typed so consumers of vLLM can get type checking

* Update py.typed

---------
Co-authored-by: aarnphm <29749331+aarnphm@users.noreply.github.com>
Co-authored-by: Zhuohan Li <zhuohan123@gmail.com>
---
 setup.py      | 1 +
 vllm/py.typed | 2 ++
 2 files changed, 3 insertions(+)
 create mode 100644 vllm/py.typed

diff --git a/setup.py b/setup.py
index 4bcd5339..daa40928 100644
--- a/setup.py
+++ b/setup.py
@@ -278,4 +278,5 @@ setuptools.setup(
     install_requires=get_requirements(),
     ext_modules=ext_modules,
     cmdclass={"build_ext": BuildExtension},
+    package_data={"vllm": ["py.typed"]},
 )
diff --git a/vllm/py.typed b/vllm/py.typed
new file mode 100644
index 00000000..33b3ad73
--- /dev/null
+++ b/vllm/py.typed
@@ -0,0 +1,2 @@
+# Marker file for PEP 561.
+# The vllm package uses inline types.