Add entrypoint to setup.py in flytekit plugins (#1120) (#1168)

Signed-off-by: Kevin Su <[email protected]> Signed-off-by: Kevin Su <[email protected]> Co-authored-by: Kevin Su <[email protected]>
flyteorg · Sep 19, 2022 · 0726203 · 0726203
1 parent c58cfb4
commit 0726203
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/plugins/flytekit-polars/setup.py b/plugins/flytekit-polars/setup.py
@@ -35,4 +35,5 @@
         "Topic :: Software Development :: Libraries",
         "Topic :: Software Development :: Libraries :: Python Modules",
     ],
+    entry_points={"flytekit.plugins": [f"{PLUGIN_NAME}=flytekitplugins.{PLUGIN_NAME}"]},
 )
diff --git a/plugins/flytekit-whylogs/setup.py b/plugins/flytekit-whylogs/setup.py
@@ -33,4 +33,5 @@
         "Topic :: Software Development :: Libraries",
         "Topic :: Software Development :: Libraries :: Python Modules",
     ],
+    entry_points={"flytekit.plugins": [f"{PLUGIN_NAME}=flytekitplugins.{PLUGIN_NAME}"]},
 )