diff --git a/examples/dueling_navix_empty.py b/examples/dueling_navix_empty.py index 1808f13..ca8a411 100644 --- a/examples/dueling_navix_empty.py +++ b/examples/dueling_navix_empty.py @@ -30,7 +30,7 @@ def main(argv): # environment env = nx.environments.Room(5, 5, 100, observation_fn=nx.observations.categorical) - env = helx.environment.to_helx(env) # type: ignore + env = helx.envs.interop.to_helx(env) # optimiser optimiser = optax.rmsprop(