rogermt commited on
Commit
8051f1f
·
verified ·
1 Parent(s): 3f3d372

v5 refactor: add constants.py

Browse files
Files changed (1) hide show
  1. neurogolf_solver/constants.py +26 -0
neurogolf_solver/constants.py ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/usr/bin/env python3
2
+ """Constants and configuration values for ARC-AGI NeuroGolf Championship."""
3
+
4
+ import numpy as np
5
+ from onnx import TensorProto
6
+
7
+ # Grid dimensions
8
+ BATCH, CH, GH, GW = 1, 10, 30, 30
9
+ GRID_SHAPE = [BATCH, CH, GH, GW]
10
+
11
+ # ONNX settings
12
+ DT = TensorProto.FLOAT
13
+ IR = 8
14
+ OPSET_VERSION = 17
15
+
16
+ # Limits
17
+ INT64_MIN = int(np.iinfo(np.int64).min)
18
+ BANNED_OPS = {'Loop', 'Scan', 'NonZero', 'Unique', 'If', 'Function'}
19
+ MAX_FILESIZE = int(1.44 * 1024 * 1024)
20
+
21
+ # Task exclusions
22
+ EXCLUDED_TASKS = {21, 55, 80, 184, 202, 366}
23
+
24
+ # ARC-GEN limits
25
+ MAX_ARCGEN_VALIDATE = 30
26
+ MAX_ARCGEN_FIT = 0