Skip to content

Commit de76bfe

Browse files
authored
added defaults for track and stop variant configuration (tensorzero#4324)
1 parent 8aa0ff5 commit de76bfe

File tree

1 file changed

+23
-0
lines changed
  • tensorzero-core/src/experimentation/track_and_stop

1 file changed

+23
-0
lines changed

tensorzero-core/src/experimentation/track_and_stop/mod.rs

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -256,14 +256,37 @@ impl Nursery {
256256
pub struct UninitializedTrackAndStopConfig {
257257
metric: String,
258258
candidate_variants: Vec<String>,
259+
#[serde(default)]
259260
fallback_variants: Vec<String>,
261+
#[serde(default = "default_min_samples_per_variant")]
260262
min_samples_per_variant: u64,
263+
#[serde(default = "default_delta")]
261264
delta: f64,
265+
#[serde(default)]
262266
epsilon: f64,
267+
#[serde(default = "default_update_period_s")]
263268
update_period_s: u64,
269+
#[serde(default = "default_min_prob")]
264270
min_prob: Option<f64>,
265271
}
266272

273+
fn default_min_samples_per_variant() -> u64 {
274+
10
275+
}
276+
277+
fn default_delta() -> f64 {
278+
0.05
279+
}
280+
281+
fn default_update_period_s() -> u64 {
282+
300
283+
}
284+
285+
#[expect(clippy::unnecessary_wraps)]
286+
fn default_min_prob() -> Option<f64> {
287+
Some(1e-6)
288+
}
289+
267290
impl UninitializedTrackAndStopConfig {
268291
pub fn load(
269292
self,

0 commit comments

Comments
 (0)