Fix sagemaker plugin (#817)

Signed-off-by: Eduardo Apolinario <eapolinario@users.noreply.github.com> Co-authored-by: Eduardo Apolinario <eapolinario@users.noreply.github.com> Signed-off-by: Eduardo Apolinario <eapolinario@users.noreply.github.com>
flyteorg · Jan 28, 2022 · 552509f · 552509f
1 parent f1af389
commit 552509f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/plugins/flytekit-aws-sagemaker/setup.py b/plugins/flytekit-aws-sagemaker/setup.py
@@ -16,7 +16,7 @@
     author_email="admin@flyte.org",
     description="AWS Plugins for flytekit",
     namespace_packages=["flytekitplugins"],
-    packages=[f"flytekitplugins.{PLUGIN_NAME}"],
+    packages=[f"flytekitplugins.{PLUGIN_NAME}", f"flytekitplugins.{PLUGIN_NAME}.models"],
     install_requires=plugin_requires,
     license="apache2",
     python_requires=">=3.7",