feat: add trial user attributes for logging and analysis

Add AttrValue enum (Float, Int, String, Bool) with From impls and
set_user_attr/user_attr/user_attrs methods on both Trial and
CompletedTrial. Attributes set during optimization are propagated
through complete_trial and prune_trial. AttrValue is re-exported
at crate root and in the prelude.
This commit is contained in:
Manuel Raimann
2026-02-11 17:18:17 +01:00
parent f67188e3a7
commit 52f3c074dc
10 changed files with 241 additions and 3 deletions
+2 -2
View File
@@ -210,7 +210,7 @@ pub use sampler::grid::GridSearchSampler;
pub use sampler::random::RandomSampler;
pub use sampler::tpe::TpeSampler;
pub use study::Study;
pub use trial::Trial;
pub use trial::{AttrValue, Trial};
pub use types::{Direction, TrialState};
/// Convenient wildcard import for the most common types.
@@ -236,6 +236,6 @@ pub mod prelude {
pub use crate::sampler::random::RandomSampler;
pub use crate::sampler::tpe::TpeSampler;
pub use crate::study::Study;
pub use crate::trial::Trial;
pub use crate::trial::{AttrValue, Trial};
pub use crate::types::Direction;
}