Added class mapping for the categorical model in the runner util. Figured out it was not included while using it in the Isaac Lab framework.