sim_tester_minimal.py

import rlgym_sim
# from CoyoteParser import CoyoteAction
from rlgym_tools.extra_state_setters.replay_setter import ReplaySetter
# from rlgym_tools.extra_state_setters.augment_setter import AugmentSetter
# from test_files.test_obs import TestObs
# from test_files.test_reward import TestReward
import numpy as np
from rlgym_sim.utils.terminal_conditions.common_conditions import GoalScoredCondition, TimeoutCondition

bad_list = []
index = {}
setter = ReplaySetter("replays/bad_1v1_doubletap_state.npy")
# obs = TestObs()
# reward = TestReward()
terminals = [GoalScoredCondition(), TimeoutCondition(100)]

env = rlgym_sim.make(tick_skip=4, spawn_opponents=True, state_setter=setter, copy_gamestate_every_step=True,
                     terminal_conditions=terminals)

total_steps = 0

while True:
    obs = env.reset()
    done = False
    steps = 0
    while not done:
        # actions = np.asarray((np.asarray([0]), np.asarray([np.random.randint(0, 373)])))
        # actions = np.asarray(np.asarray([0],))
        # actions = np.asarray([0] * 8), np.asarray([0] * 8)
        actions = np.asarray([1, 0.5, 0.5, 0.5, 0, 0, 1, 0]), np.asarray([1, 0.5, 0.5, 0.5, 0, 0, 1, 0])
        new_obs, reward, done, state = env.step(actions)
        obs = new_obs
        if np.isnan(obs).any():
            print(f"There is a nan in the obs after {steps} steps.")
            break
        steps += 1
    total_steps += steps
    print(f"completed {steps} steps. Starting new episode. Done {total_steps} total steps")