From 64bbae63005f5f3da3567d692eb905977633a1b5 Mon Sep 17 00:00:00 2001
From: Tibo De Peuter <tibo.depeuter@ugent.be>
Date: Wed, 10 Dec 2025 16:34:53 +0100
Subject: [PATCH] fixup! feat: Size arg

---
 main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/main.py b/main.py
index 7ce5560..41fdb8e 100644
--- a/main.py
+++ b/main.py
@@ -12,7 +12,7 @@ def main():
 
     match args.mode:
         case 'train':
-            size = args.size
+            size = int(args.size) if args.size else None
             if args.method == 'optuna':
                 size = 2 ** 12
                 print(f"Using size {size} for optuna (was {args.size})")