import numpy as np def returns_to_go(rewards, gamma): raise NotImplementedError def batch_returns(rewards, dones, gamma): raise NotImplementedError def normalize(returns): raise NotImplementedError