own-solver/neurogolf_solver/constants.py

#!/usr/bin/env python3
"""Constants and configuration values for ARC-AGI NeuroGolf Championship."""

import numpy as np
from onnx import TensorProto

# Grid dimensions
BATCH, CH, GH, GW = 1, 10, 30, 30
GRID_SHAPE = [BATCH, CH, GH, GW]

# ONNX settings
DT = TensorProto.FLOAT
IR = 8
OPSET_VERSION = 17

# Limits
INT64_MIN = int(np.iinfo(np.int64).min)
BANNED_OPS = {'Loop', 'Scan', 'NonZero', 'Unique', 'Script', 'Function'}
MAX_ONNX_FILESIZE = int(1.44 * 1024 * 1024)  # per .onnx file, NOT submission zip

# Task exclusions — NONE. All 400 tasks count.
EXCLUDED_TASKS = set()

# ARC-GEN limits
MAX_ARCGEN_VALIDATE = 30
MAX_ARCGEN_FIT = 0