Skip to content

Add at least one epoch to run when nsteps is fewer than 1 epoch steps #7

Add at least one epoch to run when nsteps is fewer than 1 epoch steps

Add at least one epoch to run when nsteps is fewer than 1 epoch steps #7

Workflow file for this run

name: GNS Train and Test
on:
workflow_run:
workflows: ["Build and Push to GHCR"]
types:
- completed
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: Wait for GHCR build
if: github.event_name == 'workflow_run'
run: |
echo "Build and Push to GHCR workflow completed, proceeding with GNS Train and Test"
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: Run without waiting
if: github.event_name != 'workflow_run'
run: |
echo "Build and Push to GHCR not triggered, proceeding with GNS Train and Test"
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}"
python -m gns.train
- name: Black check
run: |
black --check .