fix b_main dataset selection
This commit is contained in:
@@ -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);
|
||||||
|
@@ -21,4 +21,3 @@ namespace platform {
|
|||||||
return s;
|
return s;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@@ -23,4 +23,3 @@ namespace platform {
|
|||||||
std::locale loc;
|
std::locale loc;
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Reference in New Issue
Block a user