Use UniTree Quadruped Robots instead of ANYbotics ANYmal in AnymalTerrainTask

I want to use the unitree a1 robot for the same task as the anymalTerrain task.

Should I simply update the usd path in this script to match the unitree quadruped ? Or are there more steps ?