We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent aa0be9f commit 231cfdbCopy full SHA for 231cfdb
1 file changed
tests/tasks/test_reward_tasks.py
@@ -57,7 +57,7 @@ def reward_fn(prediction: str, reward: float) -> float:
57
58
59
def test_reward_task_x_column_from_config(simple_reward_function):
60
- """Regression for issue #67."""
+ """Ensure setting an arbitrary x_column name via the config works."""
61
df = pd.DataFrame({"my_input": ["a", "b", "c"]})
62
config = ExperimentConfig(x_column="my_input")
63
task = RewardTask(df=df, reward_function=simple_reward_function, config=config)
0 commit comments