#!/bin/bash

# Clear old files in data directory (keep .gitkeep)
rm -f games/tasks/futoshiki/data/*.jsonl

# Create 1000 samples: 4x4 grid, 2 inequality signs, 2 prefilled coordinates
python -m games.tasks.futoshiki.scripts.futoshiki_generator \
  --grid_size 4 --num_inequality_signs 2 --num_prefilled_coords 2 --num_samples 1000 \
  --output "futoshiki_4x4_ineq2_prefilled2.jsonl"

# Create 1000 samples: 4x4 grid, 4 inequality signs, 3 prefilled coordinates
python -m games.tasks.futoshiki.scripts.futoshiki_generator \
  --grid_size 4 --num_inequality_signs 4 --num_prefilled_coords 3 --num_samples 1000 \
  --output "futoshiki_4x4_ineq4_prefilled3.jsonl"

# Create 1000 samples: 5x5 grid, 7 inequality signs, 4 prefilled coordinates
python -m games.tasks.futoshiki.scripts.futoshiki_generator \
  --grid_size 5 --num_inequality_signs 7 --num_prefilled_coords 4 --num_samples 1000 \
  --output "futoshiki_5x5_ineq7_prefilled4.jsonl"

# Create 1000 samples: 5x5 grid, 10 inequality signs, 5 prefilled coordinates
python -m games.tasks.futoshiki.scripts.futoshiki_generator \
  --grid_size 5 --num_inequality_signs 10 --num_prefilled_coords 5 --num_samples 1000 \
  --output "futoshiki_5x5_ineq10_prefilled5.jsonl"