odify genome for the official release

tensorneat/problem/base.py

@@ -1,6 +1,6 @@
 from typing import Callable
 
-from utils import State, StatefulBaseClass
+from tensorneat.common import State, StatefulBaseClass
 
 
 class BaseProblem(StatefulBaseClass):

tensorneat/problem/func_fit/func_fit.py

@@ -1,7 +1,7 @@
 import jax
 import jax.numpy as jnp
 
-from utils import State
+from tensorneat.common import State
 from .. import BaseProblem
 
 

tensorneat/problem/rl_env/jumanji/jumanji_2048.py

@@ -1,7 +1,7 @@
 import jax, jax.numpy as jnp
 import jumanji
 
-from utils import State
+from tensorneat.common import State
 from ..rl_jit import RLEnv
 
 

tensorneat/problem/rl_env/rl_jit.py

@@ -4,7 +4,7 @@ import jax
 import jax.numpy as jnp
 import numpy as np
 
-from utils import State
+from tensorneat.common import State
 from .. import BaseProblem