From a0e3e1b00d6c0038eacc2cd5dccf547cf199ea89 Mon Sep 17 00:00:00 2001 From: bkb2135 <98138173+bkb2135@users.noreply.github.com> Date: Tue, 6 May 2025 23:20:41 +0100 Subject: [PATCH 1/4] Update task_registry.py --- prompting/tasks/task_registry.py | 12 +++--------- 1 file changed, 3 insertions(+), 9 deletions(-) diff --git a/prompting/tasks/task_registry.py b/prompting/tasks/task_registry.py index c16a4c83b..7f360c47c 100644 --- a/prompting/tasks/task_registry.py +++ b/prompting/tasks/task_registry.py @@ -31,22 +31,16 @@ def __hash__(self): class TaskRegistry(BaseModel): task_configs: ClassVar[list[TaskConfig]] = [ - TaskConfig(task=MSRv2Task, probability=0.05, datasets=[DDGDataset], reward_model=MSRv2RewardConfig), + TaskConfig(task=MSRv2Task, probability=0.1, datasets=[DDGDataset], reward_model=MSRv2RewardConfig), TaskConfig( task=InferenceTask, - probability=0.45, + probability=0.55, datasets=[SN13Dataset], reward_model=InferenceRewardConfig, ), - TaskConfig( - task=ProgrammingTask, - probability=0.10, - datasets=[HuggingFaceGithubDataset], - reward_model=ProgrammingRewardConfig, - ), TaskConfig( task=WebRetrievalTask, - probability=0.40, + probability=0.35, datasets=[DDGDataset], reward_model=WebRetrievalRewardConfig, ), From 639082d0d425a712008972c4ffed9fa2631eb238 Mon Sep 17 00:00:00 2001 From: richwardle Date: Wed, 7 May 2025 09:36:56 +0000 Subject: [PATCH 2/4] Precommits --- neurons/validator.py | 2 +- prompting/tasks/task_registry.py | 2 -- shared/logging/logging.py | 2 +- 3 files changed, 2 insertions(+), 4 deletions(-) diff --git a/neurons/validator.py b/neurons/validator.py index b7a5bea5f..46563abcf 100644 --- a/neurons/validator.py +++ b/neurons/validator.py @@ -12,10 +12,10 @@ import requests import torch import torch.multiprocessing as mp -import wandb from bittensor.core.extrinsics.serving import serve_extrinsic from loguru import logger +import wandb from prompting.llms.model_manager import AsyncModelScheduler, ModelManager from prompting.rewards.scoring import task_scorer diff --git a/prompting/tasks/task_registry.py b/prompting/tasks/task_registry.py index 7f360c47c..f83b0d689 100644 --- a/prompting/tasks/task_registry.py +++ b/prompting/tasks/task_registry.py @@ -5,14 +5,12 @@ from loguru import logger from pydantic import BaseModel, ConfigDict -from prompting.datasets.huggingface_github import HuggingFaceGithubDataset from prompting.datasets.random_website import DDGDataset from prompting.datasets.sn13 import SN13Dataset from prompting.rewards.reward import BaseRewardConfig from prompting.tasks.base_task import BaseTextTask from prompting.tasks.inference import InferenceRewardConfig, InferenceTask from prompting.tasks.MSRv2_task import MSRv2RewardConfig, MSRv2Task -from prompting.tasks.programming_task import ProgrammingRewardConfig, ProgrammingTask from prompting.tasks.web_retrieval import WebRetrievalRewardConfig, WebRetrievalTask from shared.base import BaseDataset diff --git a/shared/logging/logging.py b/shared/logging/logging.py index 9030ca4e4..7536464f5 100644 --- a/shared/logging/logging.py +++ b/shared/logging/logging.py @@ -4,12 +4,12 @@ from datetime import datetime, timedelta from typing import Literal -import wandb from loguru import logger from pydantic import BaseModel, ConfigDict from wandb.wandb_run import Run import prompting +import wandb from prompting.rewards.reward import WeightedRewardEvent from prompting.tasks.task_registry import TaskRegistry from shared import settings From 0a5f813bed2313c1e8df267607bcf1bad5c08a36 Mon Sep 17 00:00:00 2001 From: bkb2135 Date: Thu, 8 May 2025 10:51:37 +0000 Subject: [PATCH 3/4] Lint --- neurons/validator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/neurons/validator.py b/neurons/validator.py index 46563abcf..b7a5bea5f 100644 --- a/neurons/validator.py +++ b/neurons/validator.py @@ -12,10 +12,10 @@ import requests import torch import torch.multiprocessing as mp +import wandb from bittensor.core.extrinsics.serving import serve_extrinsic from loguru import logger -import wandb from prompting.llms.model_manager import AsyncModelScheduler, ModelManager from prompting.rewards.scoring import task_scorer From 2d85ac73049acfd8303b4fad334e63159abec7e7 Mon Sep 17 00:00:00 2001 From: bkb2135 Date: Thu, 8 May 2025 10:55:30 +0000 Subject: [PATCH 4/4] Lint more --- shared/logging/logging.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/shared/logging/logging.py b/shared/logging/logging.py index 7536464f5..9030ca4e4 100644 --- a/shared/logging/logging.py +++ b/shared/logging/logging.py @@ -4,12 +4,12 @@ from datetime import datetime, timedelta from typing import Literal +import wandb from loguru import logger from pydantic import BaseModel, ConfigDict from wandb.wandb_run import Run import prompting -import wandb from prompting.rewards.reward import WeightedRewardEvent from prompting.tasks.task_registry import TaskRegistry from shared import settings