feat(envs): rebalance poly/complex difficulties

 - make the poly-normal and poly-hard tasks slightly less difficulty
 - give complex tasks a few more moves

Author: justindujardin

libraries/mathy_python/mathy/envs/complex_simplify.py

@@ -26,7 +26,7 @@ def get_env_namespace(self) -> str:
     def max_moves_fn(
         self, problem: MathyEnvProblem, config: MathyEnvProblemArgs
     ) -> int:
-        return problem.complexity * 4
+        return problem.complexity * 5
 
     def problem_fn(self, params: MathyEnvProblemArgs) -> MathyEnvProblem:
         """Given a set of parameters to control term generation, produce

libraries/mathy_python/mathy/envs/poly_simplify.py

@@ -90,7 +90,7 @@ def problem_fn(self, params: MathyEnvProblemArgs) -> MathyEnvProblem:
             # e.g. mashing the 3rd node to commute the tree until a DF shows up in
             #      the desired position.
             noise_terms = randint(1, 4)
-            num_terms = randint(6, 10)
+            num_terms = randint(4, 8)
             scaling = uniform(0.5, 0.6)
             powers = uniform(0.35, 0.8)
             shuffle = uniform(0.35, 0.8)
@@ -104,8 +104,8 @@ def problem_fn(self, params: MathyEnvProblemArgs) -> MathyEnvProblem:
                 noise_terms=noise_terms,
             )
         elif params.difficulty == MathyEnvDifficulty.hard:
-            noise_terms = randint(1, 3)
-            num_terms = randint(8, 20)
+            noise_terms = randint(3, 7)
+            num_terms = randint(6, 10)
             scaling = uniform(0.5, 0.5)
             powers = uniform(0.15, 0.8)
             shuffle = uniform(0.35, 0.9)