From 3b670fe70de4b52b9a3f8614b42fb94fe49b3548 Mon Sep 17 00:00:00 2001 From: Lia Lenckowski Date: Sun, 3 Mar 2024 19:40:16 +0100 Subject: add hash seed option for reproducible behaviour --- src/main.rs | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) (limited to 'src/main.rs') diff --git a/src/main.rs b/src/main.rs index f9bf4fc..71e5d00 100644 --- a/src/main.rs +++ b/src/main.rs @@ -61,6 +61,10 @@ struct Args { #[arg(long, default_value = "christofides")] tsp_approx: TspAlg, + /// Seed for hashing. Random by default. + #[arg(long)] + hash_seed: Option, + images: Vec, } @@ -138,7 +142,7 @@ where } let embeds: Vec<_> = embeds.into_iter().map(|e| e.unwrap()).collect(); - let (tsp_path, total_dist) = tsp(&embeds, &args.tsp_approx); + let (tsp_path, total_dist) = tsp(&embeds, &args.tsp_approx, &args.hash_seed); Ok(( tsp_path.iter().map(|i| args.images[*i].clone()).collect(), -- cgit v1.2.3-70-g09d2