upgrade zoobot to latest version

[Tooyosi]

Upgrade zoobot to version > 2.0.

These changes aim at upgrading zoobot to the latest version. Changes made include:

• Upgrade the base cuda image to 12.1.1
• Update finetune.FinetuneableZoobotTree to match with expected params
• Change transform to test_transform from galaxy_datasets
• Import euclid_ortho_schema for new euclid workflow usecase

[yuenmichelle1]

Hey @Tooyosi
I probably am not the best person to review since my knowledge of Bajor flow is super limited. Yay to replacing that hard coded euclid schema 🥇 🎉

• I think it would be helpful to add within your PR comments, the changes that were made:
  Eg.
  Update predict_dataset to utilize new test_transform, replacing transform (Introduced in galaxy_datasets v0.0.17, galaxy_datasets used in zoobot)
  ^ It doesnt need to be this detailed, but I think super helpful for future devs to know why certain params are changed, etc.

• Probably also helpful to list what setup is already done and what setup still needs to be done to get this working in Azure Batch.
  Eg.

• TODO: Once merged, Push image to ContainerRegistry with pytorch 2.0 and create new batch pool. etc.

^ Maybe overkill but I like to see the process, plus helpful for our future selves if we have to do this kind of update again. 😄

That's at least my two cents. I'll remove myself as reviewer since my review (other than the above comments) is not going to be as fruitful as reviews from Mike et al.

[lcjohnso]

LGTM!

I'll note that pinning to python3.9 and overriding the system python3 version is OK (best to mimic Mike's recommendations for Zoobot), but using the image's default (3.10?) would be better -- prevent issues with the image AND buys us another year before Python version EOL (3.9 is EOL in late 2025).

Excited to see this up and running!

[mwalmsley]

Zoobot should work on 3.10 and above, I just haven't checked it myself. I'll make a note to do that next update.