Skip to content

Commit 0ba61d7

Browse files
committed
Support more BSBM dataset sizes
1 parent 13851fa commit 0ba61d7

File tree

1 file changed

+31
-6
lines changed

1 file changed

+31
-6
lines changed

bench/src/benchmarks/bsbm_explore.rs

Lines changed: 31 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -159,16 +159,41 @@ fn create_file_download(file: &str) -> PrepRequirement {
159159
#[derive(Clone, Copy, PartialEq, Eq, Hash, Debug, Serialize, Deserialize, ValueEnum)]
160160
pub enum BsbmDatasetSize {
161161
#[value(name = "1000")]
162-
N1000,
162+
N1_000,
163+
#[value(name = "2500")]
164+
N2_500,
163165
#[value(name = "5000")]
164-
N5000,
166+
N5_000,
167+
#[value(name = "7500")]
168+
N7_500,
169+
#[value(name = "10000")]
170+
N10_000,
171+
#[value(name = "25000")]
172+
N25_000,
173+
#[value(name = "50000")]
174+
N50_000,
175+
#[value(name = "75000")]
176+
N75_000,
177+
#[value(name = "250000")]
178+
N250_000,
179+
#[value(name = "500000")]
180+
N500_000,
165181
}
166182

167183
impl Display for BsbmDatasetSize {
168184
fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
169-
match self {
170-
BsbmDatasetSize::N1000 => f.write_str("1000"),
171-
BsbmDatasetSize::N5000 => f.write_str("5000"),
172-
}
185+
let string = match self {
186+
BsbmDatasetSize::N1_000 => "1000",
187+
BsbmDatasetSize::N2_500 => "2500",
188+
BsbmDatasetSize::N5_000 => "5000",
189+
BsbmDatasetSize::N7_500 => "7500",
190+
BsbmDatasetSize::N10_000 => "10000",
191+
BsbmDatasetSize::N25_000 => "25000",
192+
BsbmDatasetSize::N50_000 => "50000",
193+
BsbmDatasetSize::N75_000 => "75000",
194+
BsbmDatasetSize::N250_000 => "250000",
195+
BsbmDatasetSize::N500_000 => "500000",
196+
};
197+
write!(f, "{}", string)
173198
}
174199
}

0 commit comments

Comments
 (0)