def __init__(self, state_dim, hdim):
    super(SimpleValueFn, self).__init__()
    self.value_net = MLP(dims=[state_dim, *hdim, 1])
def __init__(self, state_dim, hdim):
    super(SimpleValueFn, self).__init__()
    self.value_net = MLP(dims=[state_dim, *hdim, 1])