Reorganized files to have a proper package structure

judithabk6 · Nov 28, 2023 · 667ebe5 · 667ebe5
1 parent a2d8e02
commit 667ebe5
Show file tree

Hide file tree

Showing 7 changed files with 10 additions and 6 deletions.
diff --git a/setup.py b/setup.py
@@ -4,13 +4,18 @@
     long_description = readme.read()
 
 setuptools.setup(
-    name='benchmark_mediation',
+    name='med_bench',
     version='0.1',
     author='Judith Abécassis',
     description='Benchmark mediation',
     long_description=long_description,
     long_description_content_type='text/markdown',
-    packages=setuptools.find_packages('.'),
+    # packages=setuptools.find_packages('src/*'),
+    packages=setuptools.find_packages(
+        where='src',
+        include=['med_bench*'],
+    ),
+    package_dir={"": "src"},
     install_requires=[
         'pandas>=1.2.1',
         'scikit-learn>=0.22.1',

diff --git a/src/__init__.py b/src/__init__.py
diff --git a/src/benchmark_mediation.py → src/med_bench/benchmark_mediation.py b/src/benchmark_mediation.py → src/med_bench/benchmark_mediation.py
diff --git a/src/get_estimation.py → src/med_bench/get_estimation.py b/src/get_estimation.py → src/med_bench/get_estimation.py
diff --git a/src/get_simulated_data.py → src/med_bench/get_simulated_data.py b/src/get_simulated_data.py → src/med_bench/get_simulated_data.py
diff --git a/tests/estimation/test_get_estimation.py → src/tests/estimation/test_get_estimation.py b/tests/estimation/test_get_estimation.py → src/tests/estimation/test_get_estimation.py
@@ -16,8 +16,8 @@
 import pytest
 import numpy as np
 from numpy.random import default_rng
-from med_bench.src.get_simulated_data import simulate_data
-from med_bench.src.get_estimation import get_estimation
+from med_bench.get_simulated_data import simulate_data
+from med_bench.get_estimation import get_estimation
 
 
 SMALL_ATE_TOLERANCE = 0.05

diff --git a/.../simulate_data/test_get_simulated_data.py → .../simulate_data/test_get_simulated_data.py b/.../simulate_data/test_get_simulated_data.py → .../simulate_data/test_get_simulated_data.py
@@ -18,7 +18,7 @@
 import pytest
 import numpy as np
 from numpy.random import default_rng
-from med_bench.src.get_simulated_data import simulate_data
+from med_bench.get_simulated_data import simulate_data
 
 
 PARAMETER_NAME = [