From 8fad4d7ddf050ea673ce13c2e630eee43b6d7b8e Mon Sep 17 00:00:00 2001
From: petscheit <etscheitp@gmail.com>
Date: Tue, 29 Oct 2024 12:51:28 +0100
Subject: [PATCH] chore: expose missing modules

---
 setup.py | 32 +++++++++++++++++++++++---------
 1 file changed, 23 insertions(+), 9 deletions(-)

diff --git a/setup.py b/setup.py
index 44f87145..39c24f06 100644
--- a/setup.py
+++ b/setup.py
@@ -1,8 +1,7 @@
-from setuptools import setup, find_packages
+from setuptools import setup
 from setuptools.command.install import install
 from pathlib import Path
 
-
 # Read version from version.py
 version = {}
 with open("version.py") as fp:
@@ -28,19 +27,34 @@ def run(self):
     long_description=long_description,
     long_description_content_type="text/markdown",
     version=version["__version__"],
-    packages=(
-        find_packages(where="src") +
-        find_packages(where="tools")
-    ),
+    packages=[
+        "tools.py",
+        "tools.py.providers",
+        "tools.py.providers.evm",
+        "contract_bootloader",
+        "contract_bootloader.contract_class",
+        "contract_bootloader.dryrun_syscall_memorizer_handler",
+        "contract_bootloader.dryrun_syscall_memorizer_handler.evm",
+        "contract_bootloader.memorizer",
+        "contract_bootloader.memorizer.evm",
+        "contract_bootloader.syscall_memorizer_handler",
+        "contract_bootloader.syscall_memorizer_handler.evm",
+    ],
     install_requires=requirements,
     package_dir={
-        "tools": "tools",
+        "tools.py": "tools/py",
         "contract_bootloader": "src/contract_bootloader",
     },
     zip_safe=False,
     package_data={
-        "tools": ["**/*.py"],
-        "contract_bootloader": ["*.cairo", "*/*.cairo", "*/*.py", "**/*.py", "memorizer/**/*.py"],
+        "tools.py": ["**/*.py"],
+        "contract_bootloader": [
+            "*.cairo",
+            "*/*.cairo",
+            "*/*.py",
+            "**/*.py",
+            "**/*.cairo",
+        ],
     },
     cmdclass={"install": PostInstallCommand},
 )