Fix environment to support keras-rl learner
Fix environment to support keras-rl learner

akashin/HSE_AI_Labs,akashin/HSE_AI_Labs,akashin/HSE_AI_Labs,akashin/HSE_AI_Labs,akashin/HSE_AI_Labs,akashin/HSE_AI_Labs,akashin/HSE_AI_Labs