mock wds training works locally, try on beluga

mock wds training works locally, try on beluga #242