Spaces:
Sleeping
Sleeping
Commit
·
ad3d0d5
1
Parent(s):
028ee7d
sets target prompting branch to properly integrate validator code
Browse files- requirements.txt +1 -1
- validator_wrapper.py +3 -3
requirements.txt
CHANGED
@@ -1,2 +1,2 @@
|
|
1 |
-
git+https://github.com/opentensor/prompting.git@
|
2 |
aiohttp
|
|
|
1 |
+
git+https://github.com/opentensor/prompting.git@features/move-validator-into-prompting
|
2 |
aiohttp
|
validator_wrapper.py
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
import bittensor as bt
|
2 |
-
from
|
3 |
from prompting.utils.uids import get_random_uids
|
4 |
from prompting.protocol import PromptingSynapse
|
5 |
from prompting.dendrite import DendriteResponseEvent
|
@@ -48,13 +48,13 @@ class S1ValidatorWrapper(ValidatorWrapper):
|
|
48 |
bt.logging.info(f'Calling dendrite')
|
49 |
responses = await self.validator.dendrite(
|
50 |
axons=axons,
|
51 |
-
synapse=PromptingSynapse(roles=params.roles, messages=params.
|
52 |
timeout=params.timeout,
|
53 |
)
|
54 |
|
55 |
# Encapsulate the responses in a response event (dataclass)
|
56 |
bt.logging.info(f"Creating DendriteResponseEvent:\n {responses}")
|
57 |
-
response_event = DendriteResponseEvent(responses, uids)
|
58 |
return response_event
|
59 |
|
60 |
|
|
|
1 |
import bittensor as bt
|
2 |
+
from prompting.validator import Validator
|
3 |
from prompting.utils.uids import get_random_uids
|
4 |
from prompting.protocol import PromptingSynapse
|
5 |
from prompting.dendrite import DendriteResponseEvent
|
|
|
48 |
bt.logging.info(f'Calling dendrite')
|
49 |
responses = await self.validator.dendrite(
|
50 |
axons=axons,
|
51 |
+
synapse=PromptingSynapse(roles=params.roles, messages=params.messages),
|
52 |
timeout=params.timeout,
|
53 |
)
|
54 |
|
55 |
# Encapsulate the responses in a response event (dataclass)
|
56 |
bt.logging.info(f"Creating DendriteResponseEvent:\n {responses}")
|
57 |
+
response_event = DendriteResponseEvent(responses, uids, params.timeout)
|
58 |
return response_event
|
59 |
|
60 |
|