rogermt commited on
Commit
6f2c73e
·
verified ·
1 Parent(s): e3216be

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