from typing import Any, Callable, Dict, List, NamedTuple, Tuple, Union
import torch as th

class CustomRolloutBufferSamples(NamedTuple):
    observations: th.Tensor
    actions: th.Tensor
    old_values: th.Tensor
    old_log_prob: th.Tensor
    advantages: th.Tensor
    returns: th.Tensor
    value_targets: th.Tensor
    rewards: th.Tensor

