fix training script location

[gkroiz]

Description

The nightly pax tests are failing since they use python 3.10, but the path for their training script uses a python 3.8 path (code)

Tests

Ran all of the pax nightly tests, some of them ran into scheduling issues but some also were able to train.
Example: http://shortn/_mbn1U0WDfh

Checklist

Before submitting this PR, please make sure (put X in square brackets):

• I have performed a self-review of my code.
• I have necessary comments in my code, particularly in hard-to-understand areas.
• I have run one-shot tests and provided workload links above if applicable.
• I have made or will make corresponding changes to the doc if needed.