Skip to content

Commit bd0e439

Browse files
zkytonyHarryXuancy
andauthored
Merge rocksample improvements from #84 and enable --benchmark flag (#85)
* Modify legal actions of rocksample * allow --benchmark in rocksample * pre-commit * remove gitignore * minor hash improvement --------- Co-authored-by: Chunyu Xuan <lhsxxcy@126.com>
1 parent 8018799 commit bd0e439

File tree

2 files changed

+194
-49
lines changed

2 files changed

+194
-49
lines changed

pomdp_py/__main__.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff 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

2020
if __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

0 commit comments

Comments
 (0)