#!/bin/bash
# 从 save_swe_images.sh 生成的 tar.gz 批量加载 swebench 实例镜像。

set -e

INPUT_DIR="${1:-/data1/sora/evalscope/docker/swe_images}"

if [ ! -d "$INPUT_DIR" ]; then
    echo "错误：目录不存在 $INPUT_DIR"
    exit 1
fi

TARS=("$INPUT_DIR"/swebench_batch_*.tar.gz)
TOTAL=${#TARS[@]}

echo "共找到 $TOTAL 个镜像包，开始加载..."

IDX=0
for TAR in "${TARS[@]}"; do
    IDX=$((IDX + 1))
    echo "[$IDX/$TOTAL] 加载 $TAR"
    docker load -i "$TAR"
done

echo "完成。已加载镜像数："
docker images | grep '^swebench/' | wc -l