WIP: Create Poetry package

data/__init__.py

@@ -0,0 +1,2 @@
+import model.utils as _utils
+__all__ = _utils.import_submodules(__name__).keys()

experiments/__init__.py

@@ -0,0 +1,2 @@
+import model.utils as _utils
+__all__ = _utils.import_submodules(__name__).keys()

model/__init__.py

@@ -0,0 +1,2 @@
+import model.utils as _utils
+__all__ = _utils.import_submodules(__name__).keys()

model/utils.py

@@ -3,6 +3,9 @@
 """
 
 import copy
+import sys
+import importlib
+import pkgutil
 from dataclasses import field
 from functools import partial
 
@@ -43,3 +46,17 @@ def local_variables(_locals):
 
 def default(obj):
     return field(default_factory=lambda: copy.copy(obj))
+
+
+def import_submodules(package_name):
+    """ Import all submodules of a module, recursively
+
+    :param package_name: Package name
+    :type package_name: str
+    :rtype: dict[types.ModuleType]
+    """
+    package = sys.modules[package_name]
+    return {
+        name: importlib.import_module(package_name + '.' + name)
+        for loader, name, is_pkg in pkgutil.walk_packages(package.__path__)
+    }

package/cadlabs_ethereum_economic_model/__init__.py

@@ -0,0 +1,8 @@
+"""
+CADLabs Ethereum Economic Model package & version
+"""
+__version__ = "1.1.1"
+
+import cadlabs_ethereum_economic_model.model as model
+import cadlabs_ethereum_economic_model.experiments as experiments
+import cadlabs_ethereum_economic_model.data as data

package/cadlabs_ethereum_economic_model/_utils.py

@@ -0,0 +1,17 @@
+import sys
+import importlib
+import pkgutil
+
+
+def import_submodules(package_name):
+    """ Import all submodules of a module, recursively
+
+    :param package_name: Package name
+    :type package_name: str
+    :rtype: dict[types.ModuleType]
+    """
+    package = sys.modules[package_name]
+    return {
+        name: importlib.import_module(package_name + '.' + name)
+        for loader, name, is_pkg in pkgutil.walk_packages(package.__path__)
+    }

package/cadlabs_ethereum_economic_model/data/__init__.py

@@ -0,0 +1,4 @@
+import sys
+import data
+
+sys.modules['cadlabs_ethereum_economic_model.data'] = data

package/cadlabs_ethereum_economic_model/experiments/__init__.py

@@ -0,0 +1,4 @@
+import sys
+import experiments
+
+sys.modules['cadlabs_ethereum_economic_model.experiments'] = experiments

package/cadlabs_ethereum_economic_model/model/__init__.py

@@ -0,0 +1,4 @@
+import sys
+import model
+
+sys.modules['cadlabs_ethereum_economic_model.model'] = model