From f649d6fc17cdd0f3d79965fd61ce92f8e6d8ed2a Mon Sep 17 00:00:00 2001
From: Mayuri Nehate <33225191+mayurinehate@users.noreply.github.com>
Date: Wed, 6 Sep 2023 23:08:35 +0530
Subject: [PATCH] feat(airflow-plugin): add package type information (#8795)

---
 metadata-ingestion-modules/airflow-plugin/setup.py             | 3 +++
 .../airflow-plugin/src/datahub_airflow_plugin/py.typed         | 0
 2 files changed, 3 insertions(+)
 create mode 100644 metadata-ingestion-modules/airflow-plugin/src/datahub_airflow_plugin/py.typed

diff --git a/metadata-ingestion-modules/airflow-plugin/setup.py b/metadata-ingestion-modules/airflow-plugin/setup.py
index c5bdc7ea329cd..18e605ae76ebd 100644
--- a/metadata-ingestion-modules/airflow-plugin/setup.py
+++ b/metadata-ingestion-modules/airflow-plugin/setup.py
@@ -121,6 +121,9 @@ def get_long_description():
     # Package info.
     zip_safe=False,
     python_requires=">=3.7",
+    package_data={
+        "datahub_airflow_plugin": ["py.typed"],
+    },
     package_dir={"": "src"},
     packages=setuptools.find_namespace_packages(where="./src"),
     entry_points=entry_points,
diff --git a/metadata-ingestion-modules/airflow-plugin/src/datahub_airflow_plugin/py.typed b/metadata-ingestion-modules/airflow-plugin/src/datahub_airflow_plugin/py.typed
new file mode 100644
index 0000000000000..e69de29bb2d1d