#!/bin/bash

# SPDX-FileCopyrightText: 2026 Samudra Authors
#
# SPDX-License-Identifier: Apache-2.0

# Assumes you have activated the emulator conda environment and
# set the CLUSTER_NAME and OCEAN_DATA_DIR environment variables
set -e
CONFIG_PATH=configs/eval_om4.yaml

echo 'Running local training on cluster:' $CLUSTER_NAME 'with data directory:' $OCEAN_DATA_DIR
echo 'Using config file:' $CONFIG_PATH

# current date
DATE=$(date +%Y-%m-%d)

uv run src/ocean_emulators/eval.py \
    $CONFIG_PATH \
    --ckpt_path /pscratch/sd/s/suryad/Ocean_Emulator/.LOCAL/2025-04-05-train-test/saved_nets/best_validation_ckpt.pt \
    --experiment.data_root $OCEAN_DATA_DIR \
    --data @configs/data/om4_internal.yaml \
    --experiment.name "${DATE}-om4-speed-test" \
    --save_zarr true