complete show() in brax env

2023-10-22 21:01:06 +08:00
parent 7f042e07c2
commit 15dadebd7e
11 changed files with 152 additions and 22 deletions
--- a/core/problem.py
+++ b/core/problem.py
@@ -22,7 +22,7 @@ class Problem:
    def output_shape(self):
        raise NotImplementedError

-    def show(self, randkey, state: State, act_func: Callable, params):
+    def show(self, randkey, state: State, act_func: Callable, params, *args, **kwargs):
        """
        show how a genome perform in this problem
        """