feat(airflow-plugin): add package type information (#8795)

datahub-project · Sep 6, 2023 · f649d6f · f649d6f
1 parent ac025e5
commit f649d6f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git 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