Wire 14 new transforms (object + fill + connect + compress) into default_atomic_factory

itt_solver/experiment_driver.py

@@ -24,13 +24,7 @@ def param_grid(grid_dict):
         yield dict(zip(keys, combo))
 
 def run_single(task, atomic_library, params, out_dir):
-    """
-    Run one experiment and save artifacts to out_dir.
-    This function does NOT call W&B itself to avoid recursion; call the external
-    uploader (itt_solver.wandb_runner.run_and_log_wandb) after this returns if desired.
-    """
     os.makedirs(out_dir, exist_ok=True)
-
     phi_in = initialize_potential(task['input'])
     phi_target = initialize_potential(task['target'])
     start = time.time()
@@ -54,7 +48,6 @@ def run_single(task, atomic_library, params, out_dir):
         'transform': repr(T_best),
         'states_count': len(states),
     }
-    # save best field and logs
     ts = datetime.utcnow().strftime("%Y%m%dT%H%M%SZ")
     base = f"{task.get('name','task')}_{ts}"
     np.save(os.path.join(out_dir, base + "_phi_best.npy"), phi_best)
@@ -62,7 +55,6 @@ def run_single(task, atomic_library, params, out_dir):
         json.dump(result, f, indent=2)
     with open(os.path.join(out_dir, base + "_logs.json"), "w") as f:
         json.dump(logs, f, default=str)
-
     return result
 
 def sweep(tasks, atomic_library_factory, grid, out_dir="experiments", max_runs=None):
@@ -97,13 +89,12 @@ def sweep(tasks, atomic_library_factory, grid, out_dir="experiments", max_runs=N
 def default_atomic_factory(params, task):
     """Build the default atomic library for a task.
 
-    Includes the original transforms plus the new Kronecker / mirror / upscale
-    family so the beam can express a much wider range of ARC patterns.
+    33 transforms: tiling, Kronecker, mirror, upscale, stack, object extraction,
+    fill, connect, compress, proximity, border, symmetry, gravity, color ops.
     """
     import itt_solver.transforms as tr
     from itt_solver.solver_core import tile_transform
 
-    # Capture target_shape by value to avoid late-binding closure bug
     target_h, target_w = task['target_shape'][0], task['target_shape'][1]
 
     libs = []
@@ -115,7 +106,7 @@ def default_atomic_factory(params, task):
     libs.append(tr.tile_to_target_shifted(shift=(1, 1), tile_factor=3))
     libs.append(tr.FillEnclosedHarmonic())
 
-    # --- Kronecker / self-similar (covers 007bbfb7 family) ---
+    # --- Kronecker / self-similar ---
     libs.append(tr.KroneckerSelfSimilar())
     libs.append(tr.KroneckerSelfSimilarInv())
 
@@ -124,7 +115,7 @@ def default_atomic_factory(params, task):
     libs.append(tr.MirrorTileV())
     libs.append(tr.MirrorTile4Way())
 
-    # --- upscale (2x and 3x) ---
+    # --- upscale ---
     libs.append(tr.Upscale(2))
     libs.append(tr.Upscale(3))
 
@@ -136,7 +127,25 @@ def default_atomic_factory(params, task):
     libs.append(tr.Transpose())
     libs.append(tr.CropToContent())
 
-    # --- symmetry (optional) ---
+    # --- object extraction ---
+    libs.append(tr.ExtractLargestObject())
+    libs.append(tr.ExtractSmallestObject())
+    libs.append(tr.ExtractUniqueObject())
+    libs.append(tr.ExtractMostCommonObject())
+    libs.append(tr.KeepLargestObject())
+    libs.append(tr.KeepSmallestObject())
+    libs.append(tr.SortObjectsBySize())
+
+    # --- fill / connect / compress ---
+    libs.append(tr.FillInterior())
+    libs.append(tr.ConnectSameColorH())
+    libs.append(tr.ConnectSameColorV())
+    libs.append(tr.CompressGrid())
+    libs.append(tr.RemoveBlackLines())
+    libs.append(tr.ColorByProximity())
+    libs.append(tr.DrawBorder())
+
+    # --- symmetry ---
     if params.get('use_symmetry', True):
         libs.append(tr.Rotate(1))
         libs.append(tr.Rotate(2))
@@ -144,12 +153,12 @@ def default_atomic_factory(params, task):
         libs.append(tr.Reflect('h'))
         libs.append(tr.Reflect('v'))
 
-    # --- gravity (optional) ---
+    # --- gravity ---
     if params.get('use_gravity', False):
         libs.append(tr.GravityDown())
         libs.append(tr.GravityUp())
 
-    # --- color ops (optional, for multi-color tasks) ---
+    # --- color ops ---
     if params.get('use_color_ops', False):
         libs.append(tr.InvertColors())