@torch.no_grad()
def sample_action(self, x: torch.Tensor) -> np.ndarray:
    raise NotImplementedError
@torch.no_grad()
def sample_action(self, x: torch.Tensor) -> np.ndarray:
    raise NotImplementedError