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)
{
if (argc != 2)
if (argc != 3)
throw std::runtime_error(fmt::format("Usage: {} task config.yaml", argv[0]));
if (argv[1] == std::string("unfold"))
ufo::UnfoldSolver{argv[1]}();
ufo::UnfoldSolver{argv[2]}();
else if (argv[1] == std::string("plot"))
ufo::PlotSolver{argv[1]}();
ufo::PlotSolver{argv[2]}();
else
throw std::runtime_error(fmt::format("Unknown task: {}", argv[1]));
}