fix minor bug

This commit is contained in:
陈浩南 2023-10-05 16:35:28 +08:00
parent 29cdc32ec0
commit 73cefd917c

View File

@ -2,12 +2,12 @@
int main(int argc, const char** argv) int main(int argc, const char** argv)
{ {
if (argc != 2) if (argc != 3)
throw std::runtime_error(fmt::format("Usage: {} task config.yaml", argv[0])); throw std::runtime_error(fmt::format("Usage: {} task config.yaml", argv[0]));
if (argv[1] == std::string("unfold")) if (argv[1] == std::string("unfold"))
ufo::UnfoldSolver{argv[1]}(); ufo::UnfoldSolver{argv[2]}();
else if (argv[1] == std::string("plot")) else if (argv[1] == std::string("plot"))
ufo::PlotSolver{argv[1]}(); ufo::PlotSolver{argv[2]}();
else else
throw std::runtime_error(fmt::format("Unknown task: {}", argv[1])); throw std::runtime_error(fmt::format("Unknown task: {}", argv[1]));
} }