File tree Expand file tree Collapse file tree 2 files changed +194
-49
lines changed Expand file tree Collapse file tree 2 files changed +194
-49
lines changed Original file line number Diff line number Diff line change @@ -13,12 +13,12 @@ def parse_args():
1313 available_problems
1414 ),
1515 )
16- args = parser .parse_args ()
17- return parser , args
16+ args , unknown_args = parser .parse_known_args ()
17+ return parser , args , unknown_args
1818
1919
2020if __name__ == "__main__" :
21- parser , args = parse_args ()
21+ parser , args , unknown_args = parse_args ()
2222 if args .run :
2323 if args .run .lower () == "tiger" :
2424 from pomdp_py .problems .tiger .tiger_problem import main
@@ -28,7 +28,7 @@ def parse_args():
2828 elif args .run .lower () == "rocksample" :
2929 from pomdp_py .problems .rocksample .rocksample_problem import main
3030
31- main ()
31+ main (unknown_args )
3232
3333 elif args .run .lower () == "mos" :
3434 from pomdp_py .problems .multi_object_search .problem import unittest
You can’t perform that action at this time.
0 commit comments