22 lines
914 B
Python
22 lines
914 B
Python
from app.clients.modelhub import ModelHubClient
|
|
from app.settings import Settings
|
|
|
|
|
|
def test_strategy_id_payload_field_configurable():
|
|
settings = Settings(
|
|
auth_token="a", hf_token="h", contest_api_token="c", email="e", user_id="1",
|
|
contributors="u", strategy_id="sid", contest_task_strategy_field="strategy_id"
|
|
)
|
|
payload = ModelHubClient(settings).build_contest_payload("owner/model", "hygon_k100-ai", "cfg")
|
|
assert payload["strategy_id"] == "sid"
|
|
assert payload["taskType"] == "text-generation"
|
|
|
|
|
|
def test_strategy_id_payload_can_disable_injection():
|
|
settings = Settings(
|
|
auth_token="a", hf_token="h", contest_api_token="c", email="e", user_id="1",
|
|
contributors="u", strategy_id="sid", inject_strategy_id=False
|
|
)
|
|
payload = ModelHubClient(settings).build_contest_payload("owner/model", "hygon_k100-ai", "cfg")
|
|
assert "strategyId" not in payload
|