aug_func() got an unexpected keyword argument 'obs_type'

[titanium-47]

When I run the UW-Position-Pit-Spot-v0 task using the command python scripts/reinforcement_learning/rsl_rl/train.py --task UW-Position-Pit-Spot-v0 --num_envs 1024, I get the following error
TypeError: aug_func() got an unexpected keyword argument 'obs_type'

I found that a fix for this issue is to go into UWLab/source/uwlab_tasks/uwlab_tasks/manager_based/locomotion/advance_skills/config/spot/augment.py and change aug_func to accept a keyword argument of obs_type instead of is_critic.

def aug_func(obs=None, actions=None, env=None, obs_type="policy"):
    aug_obs = None
    aug_act = None
    if obs is not None:
        if obs_type == "critic":
            aug_obs = aug_observation(obs)
        aug_obs = aug_observation(obs)
    if actions is not None:
        aug_act = aug_action(actions)
    return aug_obs, aug_act

[zoctipus]

Thanks for the bug report!

[patrickhaoy]

Thanks for reporting this Sid! I believe this issue should be fixed in PR #13. Closing this for now, but feel free to reopen it if you still see problems