Enriched default_atomic_factory with Kronecker, mirror, upscale, stack, symmetry transforms

itt_solver/experiment_driver.py

@@ -63,9 +63,6 @@ def run_single(task, atomic_library, params, out_dir):
     with open(os.path.join(out_dir, base + "_logs.json"), "w") as f:
         json.dump(logs, f, default=str)
 
-    # IMPORTANT: do not call run_and_log_wandb here to avoid recursion.
-    # If you want W&B logging, call itt_solver.wandb_runner.run_and_log_wandb(...) externally.
-
     return result
 
 def sweep(tasks, atomic_library_factory, grid, out_dir="experiments", max_runs=None):
@@ -96,20 +93,64 @@ def sweep(tasks, atomic_library_factory, grid, out_dir="experiments", max_runs=N
                 runs += 1
     return
 
-# Example atomic library factory for common ARC families
+
 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.
+    """
     import itt_solver.transforms as tr
-    from itt_solver.solver_core import tile_transform  # ensure tile_transform is available here
+    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 = []
-    # always include tile and fill
-    libs.append(tr.Transform(lambda p, _h=target_h, _w=target_w: tile_transform(p, (_h, _w)), "tile_to_target"))
+
+    # --- core tiling ---
+    libs.append(tr.Transform(
+        lambda p, _h=target_h, _w=target_w: tile_transform(p, (_h, _w)),
+        "tile_to_target"))
+    libs.append(tr.tile_to_target_shifted(shift=(1, 1), tile_factor=3))
     libs.append(tr.FillEnclosedHarmonic())
-    # optional rotations/reflections
+
+    # --- Kronecker / self-similar (covers 007bbfb7 family) ---
+    libs.append(tr.KroneckerSelfSimilar())
+    libs.append(tr.KroneckerSelfSimilarInv())
+
+    # --- mirror / kaleidoscope ---
+    libs.append(tr.MirrorTileH())
+    libs.append(tr.MirrorTileV())
+    libs.append(tr.MirrorTile4Way())
+
+    # --- upscale (2x and 3x) ---
+    libs.append(tr.Upscale(2))
+    libs.append(tr.Upscale(3))
+
+    # --- stacking ---
+    libs.append(tr.StackH(3))
+    libs.append(tr.StackV(3))
+
+    # --- structural ---
+    libs.append(tr.Transpose())
+    libs.append(tr.CropToContent())
+
+    # --- symmetry (optional) ---
     if params.get('use_symmetry', True):
         libs.append(tr.Rotate(1))
+        libs.append(tr.Rotate(2))
+        libs.append(tr.Rotate(3))
         libs.append(tr.Reflect('h'))
+        libs.append(tr.Reflect('v'))
+
+    # --- gravity (optional) ---
+    if params.get('use_gravity', False):
+        libs.append(tr.GravityDown())
+        libs.append(tr.GravityUp())
+
+    # --- color ops (optional, for multi-color tasks) ---
+    if params.get('use_color_ops', False):
+        libs.append(tr.InvertColors())
+
     return libs