Я планирую обучить агента Mario играть самостоятельно, используя DQN.Но я также хочу иметь возможность контролировать агента, например, если он застрянет на трубе, а затем позволить ему продолжать играть самому.Как мне извлечь конкретное состояние, где ему нужна помощь?