preference-lab / __init__.py
Sibam
PreferenceLab OpenEnv environment for RLHF preference simulation
cdf485e
Raw
History Blame Contribute Delete
593 Bytes
"""
PreferenceLab: An OpenEnv environment simulating the RLHF
preference data collection pipeline.
Usage:
from preference_lab import PreferenceLabEnv, PairwiseAction, LikertAction, ConsistencyAction
"""
from client import PreferenceLabEnv
from models import (
ConsistencyAction,
ConsistencyObservation,
LikertAction,
LikertObservation,
PairwiseAction,
PairwiseObservation,
)
__all__ = [
"PreferenceLabEnv",
"PairwiseAction",
"PairwiseObservation",
"LikertAction",
"LikertObservation",
"ConsistencyAction",
"ConsistencyObservation",
]