[RLlib] Move bandit example scripts into examples folder. (#21949)

This commit is contained in:
Jun Gong 2022-02-02 00:20:47 -08:00 committed by GitHub
parent 3d449d4f71
commit a55258eb9c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 25 additions and 24 deletions

View file

@ -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
#