train.yml

name: GNS Train and Test

on:
  push:
  pull_request:

jobs:
  gns:
    runs-on: ubuntu-latest
    container:
      image: ghcr.io/geoelements/gns:config

    steps:
    - name: Checkout repository
      uses: actions/checkout@v4

    - name: Train & Test
      run: |
        TMP_DIR="./gns-sample"
        DATASET_NAME="WaterDropSample"
        git clone https://github.com/geoelements/gns-sample
        mkdir -p ${TMP_DIR}/${DATASET_NAME}/models/
        mkdir -p ${TMP_DIR}/${DATASET_NAME}/rollout/
        DATA_PATH="${TMP_DIR}/${DATASET_NAME}/dataset/"
        MODEL_PATH="${TMP_DIR}/${DATASET_NAME}/models/"
        ROLLOUT_PATH="${TMP_DIR}/${DATASET_NAME}/rollout/"
        pytest test/
        echo "Test paths: ${DATA_PATH} ${MODEL_PATH}"
        ls
        python -m gns.train
        echo "Predict rollout"
        ls ./gns-sample/WaterDropSample/models/

    - name: Black check
      run: |
        conda install -c conda-forge black -y
        black --check .