File size: 1,078 Bytes
07423df |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
import pytest
from llm_studio.python_configs.text_dpo_modeling_config import (
ConfigDPODataset,
ConfigProblemBase,
)
from llm_studio.src.utils.utils import PatchedAttribute
def test_patched_attribute():
cfg = ConfigProblemBase(
dataset=ConfigDPODataset(
prompt_column=("prompt_column",),
answer_column="answer_column",
rejected_answer_column="rejected_answer_column",
parent_id_column="None",
)
)
with PatchedAttribute(cfg.dataset, "answer_column", "chosen_response"):
assert cfg.dataset.answer_column == "chosen_response"
with PatchedAttribute(cfg.dataset, "answer_column", "new_answer_column"):
assert cfg.dataset.answer_column == "new_answer_column"
assert cfg.dataset.answer_column == "answer_column"
with PatchedAttribute(cfg.dataset, "new_property", "new_value"):
assert cfg.dataset.new_property == "new_value" # type: ignore[attr-defined]
with pytest.raises(AttributeError):
cfg.dataset.new_property # type: ignore[attr-defined]
|