fix: Ensure that __init__ files exist in proto dirs (feast-dev#2433)

Signed-off-by: Achal Shah <achals@gmail.com>
twitter-forks · Mar 22, 2022 · 9b94f7b · 9b94f7b
1 parent bb82741
commit 9b94f7b
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/sdk/python/setup.py b/sdk/python/setup.py
@@ -224,6 +224,11 @@ def run(self):
             with open(f"{self.python_folder}/feast/{sub_folder}/__init__.py", 'w'):
                 pass
 
+        with open(f"{self.python_folder}/__init__.py", 'w'):
+            pass
+        with open(f"{self.python_folder}/feast/__init__.py", 'w'):
+            pass
+
         for path in Path("feast/protos").rglob("*.py"):
             for folder in self.sub_folders:
                 # Read in the file