From 6ca7612d89508abb4091bcdd56758298681bfa38 Mon Sep 17 00:00:00 2001 From: Matthew Smith Date: Fri, 23 Aug 2019 01:19:03 -0700 Subject: [PATCH 1/2] Add modules to lazy load list --- flytekit/plugins/__init__.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/flytekit/plugins/__init__.py b/flytekit/plugins/__init__.py index ca5d210c1b..46ffdfdbad 100644 --- a/flytekit/plugins/__init__.py +++ b/flytekit/plugins/__init__.py @@ -5,7 +5,8 @@ "spark", ["pyspark>=2.4.0,<3.0.0"], [ - "pyspark" + "pyspark", + "flyteidl.plugins.spark_pb2", ] ) @@ -15,6 +16,7 @@ [ "k8s.io.api.core.v1.generated_pb2", "k8s.io.apimachinery.pkg.api.resource.generated_pb2", + "flyteidl.plugins.sidecar_pb2", ] ) From be6231a6ee6c9510aa0cb380fd4990b3b8a6ef1c Mon Sep 17 00:00:00 2001 From: Matthew Smith Date: Fri, 23 Aug 2019 01:28:18 -0700 Subject: [PATCH 2/2] Only needed for sidecar, not spark --- flytekit/plugins/__init__.py | 1 - 1 file changed, 1 deletion(-) diff --git a/flytekit/plugins/__init__.py b/flytekit/plugins/__init__.py index 46ffdfdbad..361346f960 100644 --- a/flytekit/plugins/__init__.py +++ b/flytekit/plugins/__init__.py @@ -6,7 +6,6 @@ ["pyspark>=2.4.0,<3.0.0"], [ "pyspark", - "flyteidl.plugins.spark_pb2", ] )