From d56588dba8a944feeafdf23e313cd566df30fcf1 Mon Sep 17 00:00:00 2001 From: arnupretorius Date: Mon, 4 Sep 2023 16:00:56 +0200 Subject: [PATCH] fix: registry --- matrax/__init__.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/matrax/__init__.py b/matrax/__init__.py index 67b4bd9..51c1fc9 100644 --- a/matrax/__init__.py +++ b/matrax/__init__.py @@ -12,7 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -import jax.numpy as jnp from jumanji.registration import make, register from jumanji.version import __version__ @@ -34,7 +33,7 @@ ) register( id=f"Penalty-{penalty_value}-stateful-v0", - entry_point="matrixgames.games:MatrixGame", + entry_point="matrax:MatrixGame", kwargs={ "payoff_matrix": payoff_matrix, "keep_state": True, @@ -53,7 +52,7 @@ ) register( id="Climbing-stateful-v0", - entry_point="matrixgames.games:MatrixGame", + entry_point="matrax:MatrixGame", kwargs={ "payoff_matrix": climbing_game, "keep_state": True, @@ -73,7 +72,7 @@ ) register( id=f"NoConflict-{_id}-stateful-v0", - entry_point="matrixgames.games:MatrixGame", + entry_point="matrax:MatrixGame", kwargs={ "payoff_matrix": payoff_matrix, "keep_state": True, @@ -93,7 +92,7 @@ ) register( f"Conflict-{_id}-stateful-v0", - entry_point="matrixgames.games:MatrixGame", + entry_point="matrix:MatrixGame", kwargs={ "payoff_matrix": payoff_matrix, "keep_state": True,