Add file size check: reject oversized .onnx before validation, --strict_size stops run on violation

neurogolf_solver/validators.py

@@ -1,14 +1,22 @@
 #!/usr/bin/env python3
 """Model validation utilities."""
 
+import os
 import numpy as np
 import onnxruntime as ort
 from .data_loader import to_onehot
-from .constants import MAX_ARCGEN_VALIDATE
+from .constants import MAX_ARCGEN_VALIDATE, MAX_ONNX_FILESIZE
 
 
 def validate(path, td, providers):
-    """Validate ONNX model against task data."""
+    """Validate ONNX model against task data.
+    Returns False immediately if file exceeds 1.44MB."""
+    # Size check FIRST — before wasting time on inference
+    try:
+        if os.path.getsize(path) > MAX_ONNX_FILESIZE:
+            return False
+    except OSError:
+        return False
     try:
         opts = ort.SessionOptions()
         opts.log_severity_level = 3
@@ -33,6 +41,9 @@ def validate(path, td, providers):
 
 def validate_raw(raw_bytes, td, providers):
     """Validate ONNX model from raw bytes."""
+    # Size check
+    if len(raw_bytes) > MAX_ONNX_FILESIZE:
+        return False
     try:
         opts = ort.SessionOptions()
         opts.log_severity_level = 3