fix: Ensure that __init__ files exist in proto dirs

Signed-off-by: Achal Shah <[email protected]>
feast-dev · Mar 22, 2022 · 6e4dd62 · 6e4dd62
1 parent e0041f1
commit 6e4dd62
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