fix: remove absolute path in module (#44)

alonfnt · Nov 27, 2024 · fc1c080 · fc1c080
1 parent f2ac00e
commit fc1c080
Show file tree

Hide file tree

Showing 8 changed files with 9 additions and 9 deletions.
diff --git a/bayex/__init__.py b/bayex/__init__.py
@@ -1,8 +1,8 @@
-from bayex.src.optimizer import Optimizer, OptimizerState
-from bayex.src import domain
+from .optimizer import Optimizer, OptimizerState
+from . import domain
 
 
-__version__ = "0.2.0"
+__version__ = "0.2.1c"
 
 __all__ = [
         "Optimizer",

diff --git a/bayex/src/acq.py → bayex/acq.py b/bayex/src/acq.py → bayex/acq.py
@@ -2,7 +2,7 @@
 import jax.numpy as jnp
 from jax.scipy.stats import norm
 
-from bayex.src.gp import GPParams, predict
+from bayex.gp import GPParams, predict
 
 
 def expected_improvement(

diff --git a/bayex/src/domain.py → bayex/domain.py b/bayex/src/domain.py → bayex/domain.py
diff --git a/bayex/src/gp.py → bayex/gp.py b/bayex/src/gp.py → bayex/gp.py
diff --git a/bayex/src/gp_test.py → bayex/gp_test.py b/bayex/src/gp_test.py → bayex/gp_test.py
@@ -1,6 +1,6 @@
 import jax.numpy as jnp
 import pytest
-from bayex.src.gp import GPParams, gaussian_process, exp_quadratic, cov
+from bayex.gp import GPParams, gaussian_process, exp_quadratic, cov
 
 @pytest.mark.parametrize("x1, x2, mask, expected", [
     (jnp.array([0]), jnp.array([0]), 1, 1),

diff --git a/bayex/src/optimizer.py → bayex/optimizer.py b/bayex/src/optimizer.py → bayex/optimizer.py
@@ -5,8 +5,8 @@
 import jax.numpy as jnp
 import numpy as np
 
-import bayex.src.acq as boacq
-from bayex.src.gp import GPParams, GPState, posterior_fit
+import bayex.acq as boacq
+from bayex.gp import GPParams, GPState, posterior_fit
 
 
 class OptimizerState(NamedTuple):

diff --git a/bayex/src/optimizer_test.py → bayex/optimizer_test.py b/bayex/src/optimizer_test.py → bayex/optimizer_test.py
@@ -29,7 +29,7 @@ def f(x):
     assert opt_state.best_params['x'] == params['x'][np.argmax(ys)]
 
     assert type(opt_state) == bayex.OptimizerState
-    assert type(opt_state.gp_state) == bayex.src.gp.GPState # pyright: ignore
+    assert type(opt_state.gp_state) == bayex.gp.GPState # pyright: ignore
 
     assert opt_state.params['x'].shape == (10,)
     assert opt_state.ys.shape == (10,)

diff --git a/pyproject.toml b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "bayex"
-version = "0.2.0"
+version = "0.2.1c"
 description = "Minimal Bayesian Optimization Implementation with Gaussian Processes written in JAX."
 readme = "README.md"
 requires-python = ">=3.9"