fix b_main dataset selection

This commit is contained in:
2024-03-19 17:37:32 +01:00
parent 634ea36169
commit 673a41fc4d
3 changed files with 4 additions and 4 deletions

View File

@@ -18,9 +18,11 @@ void manageArguments(argparse::ArgumentParser& program)
auto datasets = platform::Datasets(false, platform::Paths::datasets()); auto datasets = platform::Datasets(false, platform::Paths::datasets());
program.add_argument("-d", "--dataset") program.add_argument("-d", "--dataset")
.help("Dataset file name: " + datasets.toString()) .help("Dataset file name: " + datasets.toString())
.default_value("all")
.action([](const std::string& value) { .action([](const std::string& value) {
auto datasets = platform::Datasets(false, platform::Paths::datasets()); auto datasets = platform::Datasets(false, platform::Paths::datasets());
static const std::vector<std::string> choices_datasets(datasets.getNames()); static std::vector<std::string> choices_datasets(datasets.getNames());
choices_datasets.push_back("all");
if (find(choices_datasets.begin(), choices_datasets.end(), value) != choices_datasets.end()) { if (find(choices_datasets.begin(), choices_datasets.end(), value) != choices_datasets.end()) {
return value; return value;
} }
@@ -102,7 +104,7 @@ int main(int argc, char** argv)
exit(1); exit(1);
} }
auto datasets = platform::Datasets(discretize_dataset, platform::Paths::datasets()); auto datasets = platform::Datasets(discretize_dataset, platform::Paths::datasets());
if (file_name != "") { if (file_name != "all") {
if (!datasets.isDataset(file_name)) { if (!datasets.isDataset(file_name)) {
cerr << "Dataset " << file_name << " not found" << std::endl; cerr << "Dataset " << file_name << " not found" << std::endl;
exit(1); exit(1);

View File

@@ -21,4 +21,3 @@ namespace platform {
return s; return s;
} }
} }

View File

@@ -23,4 +23,3 @@ namespace platform {
std::locale loc; std::locale loc;
}; };
} }