Skip to content

Commit 24bd1c1

Browse files
authored
Relocate priors (#238)
This PR moves our priors from `baybe.kernels.priors` to `baybe.priors` since they can be used in general non-kernel contexts, for instance, as in #231.
2 parents 09243ce + 7d31c4d commit 24bd1c1

File tree

10 files changed

+12
-11
lines changed

10 files changed

+12
-11
lines changed

baybe/kernels/base.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77

88
from attrs import define
99

10-
from baybe.kernels.priors.base import Prior
10+
from baybe.priors.base import Prior
1111
from baybe.serialization.core import (
1212
converter,
1313
get_base_structure_hook,

baybe/kernels/basic.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
from attrs.validators import optional as optional_v
99

1010
from baybe.kernels.base import Kernel
11-
from baybe.kernels.priors.base import Prior
11+
from baybe.priors.base import Prior
1212
from baybe.utils.conversion import fraction_to_float
1313
from baybe.utils.validation import finite_float
1414

baybe/kernels/priors/__init__.py baybe/priors/__init__.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
1-
"""Available priors."""
1+
"""Prior distributions."""
22

3-
from baybe.kernels.priors.basic import (
3+
from baybe.priors.basic import (
44
GammaPrior,
55
HalfCauchyPrior,
66
HalfNormalPrior,
File renamed without changes.

baybe/kernels/priors/basic.py baybe/priors/basic.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,11 @@
1-
"""Priors that can be used for kernels."""
1+
"""A collection of common prior distributions."""
2+
23
from typing import Any
34

45
from attrs import define, field
56
from attrs.validators import gt
67

7-
from baybe.kernels.priors.base import Prior
8+
from baybe.priors.base import Prior
89
from baybe.utils.validation import finite_float
910

1011

baybe/surrogates/gaussian_process.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88

99
from baybe.kernels import MaternKernel, ScaleKernel
1010
from baybe.kernels.base import Kernel
11-
from baybe.kernels.priors import GammaPrior
11+
from baybe.priors import GammaPrior
1212
from baybe.searchspace import SearchSpace
1313
from baybe.surrogates.base import Surrogate
1414

tests/conftest.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,6 @@
2828
)
2929
from baybe.exceptions import OptionalImportError
3030
from baybe.kernels import MaternKernel
31-
from baybe.kernels.priors import GammaPrior
3231
from baybe.objectives.desirability import DesirabilityObjective
3332
from baybe.objectives.single import SingleTargetObjective
3433
from baybe.parameters import (
@@ -39,6 +38,7 @@
3938
SubstanceEncoding,
4039
TaskParameter,
4140
)
41+
from baybe.priors import GammaPrior
4242
from baybe.recommenders.meta.base import MetaRecommender
4343
from baybe.recommenders.meta.sequential import (
4444
SequentialMetaRecommender,

tests/hypothesis_strategies/priors.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

33
import hypothesis.strategies as st
44

5-
from baybe.kernels.priors import (
5+
from baybe.priors import (
66
GammaPrior,
77
HalfCauchyPrior,
88
HalfNormalPrior,

tests/serialization/test_prior_serialization.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

33
from hypothesis import given
44

5-
from baybe.kernels.priors.base import Prior
5+
from baybe.priors.base import Prior
66
from tests.hypothesis_strategies.priors import priors
77

88

tests/test_iterations.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
from baybe.acquisition.base import AcquisitionFunction
88
from baybe.kernels import MaternKernel, ScaleKernel
9-
from baybe.kernels.priors import (
9+
from baybe.priors import (
1010
GammaPrior,
1111
HalfCauchyPrior,
1212
HalfNormalPrior,

0 commit comments

Comments
 (0)