Wire new solvers into registry (gravity, edge, mode_fill)
Browse files
neurogolf_solver/solvers/solver_registry.py
CHANGED
|
@@ -9,6 +9,7 @@ from .geometric import s_flip, s_rotate, s_shift, s_fixed_crop, s_gravity
|
|
| 9 |
from .tiling import (s_tile, s_upscale, s_kronecker, s_nonuniform_scale, s_diagonal_tile,
|
| 10 |
s_mirror_h, s_mirror_v, s_quad_mirror, s_concat, s_concat_enhanced,
|
| 11 |
s_spatial_gather, s_varshape_spatial_gather)
|
|
|
|
| 12 |
from .conv import solve_conv_fixed, solve_conv_variable, solve_conv_diffshape, solve_conv_var_diff
|
| 13 |
from ..data_loader import get_exs, fixed_shapes
|
| 14 |
from ..validators import validate
|
|
@@ -36,6 +37,9 @@ ANALYTICAL_SOLVERS = [
|
|
| 36 |
('fixed_crop', s_fixed_crop),
|
| 37 |
('spatial_gather', s_spatial_gather),
|
| 38 |
('varshape_spatial_gather', s_varshape_spatial_gather),
|
|
|
|
|
|
|
|
|
|
| 39 |
]
|
| 40 |
|
| 41 |
|
|
|
|
| 9 |
from .tiling import (s_tile, s_upscale, s_kronecker, s_nonuniform_scale, s_diagonal_tile,
|
| 10 |
s_mirror_h, s_mirror_v, s_quad_mirror, s_concat, s_concat_enhanced,
|
| 11 |
s_spatial_gather, s_varshape_spatial_gather)
|
| 12 |
+
from .new_solvers import s_gravity_unrolled, s_edge_detect, s_mode_fill
|
| 13 |
from .conv import solve_conv_fixed, solve_conv_variable, solve_conv_diffshape, solve_conv_var_diff
|
| 14 |
from ..data_loader import get_exs, fixed_shapes
|
| 15 |
from ..validators import validate
|
|
|
|
| 37 |
('fixed_crop', s_fixed_crop),
|
| 38 |
('spatial_gather', s_spatial_gather),
|
| 39 |
('varshape_spatial_gather', s_varshape_spatial_gather),
|
| 40 |
+
('gravity_unrolled', s_gravity_unrolled),
|
| 41 |
+
('edge_detect', s_edge_detect),
|
| 42 |
+
('mode_fill', s_mode_fill),
|
| 43 |
]
|
| 44 |
|
| 45 |
|