diff --git a/sdk/python/setup.py b/sdk/python/setup.py
index 351f32b71b..fea83267ce 100644
--- 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