Update workflow

terryyylim · terryyylim · commit 01e228b49fbf · 2021-02-11T10:42:57.000+08:00
Signed-off-by: Terence Lim &lt;terencelimxp@gmail.com&gt;
diff --git a/.prow.yaml b/.prow.yaml
@@ -108,7 +108,7 @@ postsubmits:
       - sh
       - -c
       - |
-        make compile-protos-python && infra/scripts/publish-python-sdk.sh \
+        make package-protos && make compile-protos-python && infra/scripts/publish-python-sdk.sh \
           --directory-path sdk/python --repository pypi
       volumeMounts:
       - name: pypirc
diff --git a/Makefile b/Makefile
@@ -64,6 +64,9 @@ build-java-no-tests:
 install-python-ci-dependencies:
 	pip install --no-cache-dir -r sdk/python/requirements-ci.txt
 
+package-protos:
+	cp -r ${ROOT_DIR}/protos ${ROOT_DIR}/sdk/python/feast/protos
+
 compile-protos-python: install-python-ci-dependencies
 	@$(foreach dir,$(PROTO_TYPE_SUBDIRS),cd ${ROOT_DIR}/protos; python -m grpc_tools.protoc -I. --python_out=../sdk/python/ --mypy_out=../sdk/python/ feast/$(dir)/*.proto;)
 	@$(foreach dir,$(PROTO_SERVICE_SUBDIRS),cd ${ROOT_DIR}/protos; python -m grpc_tools.protoc -I. --grpc_python_out=../sdk/python/ feast/$(dir)/*.proto;)
diff --git a/sdk/python/setup.py b/sdk/python/setup.py
@@ -94,4 +94,11 @@
     entry_points={"console_scripts": ["feast=feast.cli:cli"]},
     use_scm_version={"root": "../..", "relative_to": __file__, "tag_regex": TAG_REGEX},
     setup_requires=["setuptools_scm"],
+    package_data={
+        "": [
+            "protos/feast/**/*.proto",
+            "protos/feast/third_party/grpc/health/v1/*.proto",
+            "protos/tensorflow_metadata/proto/v0/*.proto",
+        ],
+    },
 )
