mirror of
https://github.com/vale981/ray
synced 2025-03-05 10:01:43 -05:00
[RLlib] Move bandit example scripts into examples folder. (#21949)
This commit is contained in:
parent
3d449d4f71
commit
a55258eb9c
4 changed files with 25 additions and 24 deletions
49
rllib/BUILD
49
rllib/BUILD
|
@ -596,30 +596,6 @@ py_test(
|
|||
srcs = ["agents/bandit/tests/test_bandits.py"],
|
||||
)
|
||||
|
||||
py_test(
|
||||
name = "agents/bandit/tests/lin_ts_train_wheel_env",
|
||||
main = "agents/bandit/tests/LinTS_train_wheel_env.py",
|
||||
tags = ["team:ml", "trainers_dir"],
|
||||
size = "small",
|
||||
srcs = ["agents/bandit/tests/LinTS_train_wheel_env.py"],
|
||||
)
|
||||
|
||||
py_test(
|
||||
name = "bandit/tune_lin_ts_train_wheel_env",
|
||||
main = "agents/bandit/tests/tune_LinTS_train_wheel_env.py",
|
||||
tags = ["team:ml", "trainers_dir"],
|
||||
size = "small",
|
||||
srcs = ["agents/bandit/tests/tune_LinTS_train_wheel_env.py"],
|
||||
)
|
||||
|
||||
py_test(
|
||||
name = "bandit/tune_lin_ucb_train_recommendation",
|
||||
main = "agents/bandit/tests/tune_LinUCB_train_recommendation.py",
|
||||
tags = ["team:ml", "trainers_dir"],
|
||||
size = "small",
|
||||
srcs = ["agents/bandit/tests/tune_LinUCB_train_recommendation.py"],
|
||||
)
|
||||
|
||||
# CQLTrainer
|
||||
py_test(
|
||||
name = "test_cql",
|
||||
|
@ -2766,6 +2742,31 @@ py_test(
|
|||
args = ["--as-test", "--framework=torch", "--stop-reward=7", "--run=PG"]
|
||||
)
|
||||
|
||||
|
||||
py_test(
|
||||
name = "examples/bandit/lin_ts_train_wheel_env",
|
||||
main = "examples/bandit/lin_ts_train_wheel_env.py",
|
||||
tags = ["team:ml", "examples"],
|
||||
size = "small",
|
||||
srcs = ["examples/bandit/lin_ts_train_wheel_env.py"],
|
||||
)
|
||||
|
||||
py_test(
|
||||
name = "examples/bandit/tune_lin_ts_train_wheel_env",
|
||||
main = "examples/bandit/tune_lin_ts_train_wheel_env.py",
|
||||
tags = ["team:ml", "examples"],
|
||||
size = "small",
|
||||
srcs = ["examples/bandit/tune_lin_ts_train_wheel_env.py"],
|
||||
)
|
||||
|
||||
py_test(
|
||||
name = "examples/bandit/tune_lin_ucb_train_recommendation",
|
||||
main = "examples/bandit/tune_lin_ucb_train_recommendation.py",
|
||||
tags = ["team:ml", "examples", ],
|
||||
size = "small",
|
||||
srcs = ["examples/bandit/tune_lin_ucb_train_recommendation.py"],
|
||||
)
|
||||
|
||||
# --------------------------------------------------------------------
|
||||
# examples/documentation directory
|
||||
#
|
||||
|
|
Loading…
Add table
Reference in a new issue