Update solver_registry.py to include Wave 2 solvers (17 new analytical solvers)

own-solver/neurogolf_solver/solvers/solver_registry.py

@@ -14,6 +14,11 @@ from .edge import s_edge_detect
 from .mode import s_mode_fill
 from .wave1 import (s_downsample_stride, s_symmetry_complete, s_extract_inner,
                     s_add_border, s_sparse_fill, s_channel_filter)
+from .wave2 import (s_overlay_constant, s_bbox_crop, s_row_mode_fill, s_col_mode_fill,
+                    s_fill_bg_with_mode, s_pad_align, s_multi_stamp, s_diagonal_flip,
+                    s_invert_colors, s_majority_color_fill, s_border_extract,
+                    s_interior_fill, s_repeat_row, s_repeat_col, s_swap_two_colors,
+                    s_max_pool_downsample, s_crop_paste)
 from .conv import solve_conv_fixed, solve_conv_variable, solve_conv_diffshape, solve_conv_var_diff
 from ..data_loader import get_exs, fixed_shapes
 from ..validators import validate
@@ -45,12 +50,31 @@ ANALYTICAL_SOLVERS = [
     ('gravity_unrolled', s_gravity_unrolled),
     ('edge_detect', s_edge_detect),
     ('mode_fill', s_mode_fill),
+    # Wave 1 solvers
     ('downsample_stride', s_downsample_stride),
     ('symmetry_complete', s_symmetry_complete),
     ('extract_inner', s_extract_inner),
     ('add_border', s_add_border),
     ('sparse_fill', s_sparse_fill),
     ('channel_filter', s_channel_filter),
+    # Wave 2 solvers (new)
+    ('swap_two_colors', s_swap_two_colors),
+    ('invert_colors', s_invert_colors),
+    ('diagonal_flip', s_diagonal_flip),
+    ('overlay_constant', s_overlay_constant),
+    ('bbox_crop', s_bbox_crop),
+    ('row_mode_fill', s_row_mode_fill),
+    ('col_mode_fill', s_col_mode_fill),
+    ('fill_bg_with_mode', s_fill_bg_with_mode),
+    ('pad_align', s_pad_align),
+    ('repeat_row', s_repeat_row),
+    ('repeat_col', s_repeat_col),
+    ('border_extract', s_border_extract),
+    ('interior_fill', s_interior_fill),
+    ('majority_color_fill', s_majority_color_fill),
+    ('max_pool_downsample', s_max_pool_downsample),
+    ('crop_paste', s_crop_paste),
+    ('multi_stamp', s_multi_stamp),
 ]