move all math to math folder

Author: rizar

conf/actor/math.yaml

@@ -1,7 +1,7 @@
 log_each_n_secs: 10
 llm_max_rollouts: 128
 rollout_workers: 1
-rollout_policy: pipelinerl.math_rollouts.generate_math_rollout
+rollout_policy: pipelinerl.math.rollouts.generate_math_rollout
 discount_factor: 1
 system_prompt: Please reason step by step, and put your final answer within \boxed{}.
 task_template: |-

pipelinerl/entrypoints/verifier.py

@@ -1,7 +1,7 @@
 import hydra
 from omegaconf import DictConfig
 
-from pipelinerl.verifier_api import run_verifier
+from pipelinerl.math.verifier_api import run_verifier
 from pipelinerl.utils import better_crashing
 
 

pipelinerl/math_rollouts.py renamed to pipelinerl/math/rollouts.py

@@ -8,7 +8,7 @@
 
 from pipelinerl.async_llm import llm_async_generate
 from pipelinerl.finetune.data import MASKED_TOKEN_ID
-from pipelinerl.verifier_api import verify_answer_rpc
+from pipelinerl.math.verifier_api import verify_answer_rpc
 
 
 class RewardTable(BaseModel):

pipelinerl/verifier_api.py renamed to pipelinerl/math/verifier_api.py

@@ -20,7 +20,7 @@
 import wandb
 from pipelinerl.finetune.logging_ import flatten_dict_config, init_wandb
 from pipelinerl.load_datasets import load_datasets
-from pipelinerl.math_rollouts import RolloutResult
+from pipelinerl.math.rollouts import RolloutResult
 from pipelinerl.shared_memory_array import SharedMemoryArray
 from pipelinerl.state import TrainerState
 from pipelinerl.streams import (
@@ -30,7 +30,7 @@
     set_streams_backend,
     write_to_streams,
 )
-from pipelinerl.verifier_api import wait_for_verifier
+from pipelinerl.math.verifier_api import wait_for_verifier
 
 from .utils import (
     always_or_never_success_stats,