steps:
  "train":
    image: nvcr.io/nvidia/tensorflow:23.10-tf2-py3
    commands:
      - echo "starting python script"
      - zip
      - python run.py