Skip to content

Commit

Permalink
Fix test.
Browse files Browse the repository at this point in the history
  • Loading branch information
frozenlib committed Jan 22, 2025
1 parent b545895 commit f0f801b
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion tests/mlflow_client/mod.rs
Original file line number Diff line number Diff line change
Expand Up @@ -321,7 +321,7 @@ fn log_batch() -> Result<()> {
let c = s.mlflow_client();
let r0 = c.create_experiment("abc", Default::default())?;
let r1 = c.create_run(&r0.experiment_id, "", Default::default())?;
let metrics = vec![
let mut metrics = vec![
Metric {
key: "m1".to_string(),
value: 1.0,
Expand Down Expand Up @@ -357,6 +357,8 @@ fn log_batch() -> Result<()> {
];
c.log_batch(&r1.run.info.run_id, &metrics, &params, &tags)?;
let mut r2 = c.get_run(&r1.run.info.run_id)?;
r2.run.data.params.sort();
metrics.sort();
assert_eq!(&r2.run.data.metrics, &metrics);
r2.run.data.params.sort();
params.sort();
Expand Down

0 comments on commit f0f801b

Please sign in to comment.