From 30a09bd66c46f29e1463287982bd03afc7ecda19 Mon Sep 17 00:00:00 2001
From: Avnish Narayan <38871737+avnishn@users.noreply.github.com>
Date: Tue, 22 Nov 2022 00:23:06 -0800
Subject: [PATCH] Bump tests to medium (#30548)

---
 rllib/BUILD | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/rllib/BUILD b/rllib/BUILD
index 2e0ec1ce1e4a..90ea726b6c21 100644
--- a/rllib/BUILD
+++ b/rllib/BUILD
@@ -1311,7 +1311,7 @@ py_test(
 py_test(
     name = "test_dqn_frozenlake_v1",
     main = "train.py", srcs = ["train.py"],
-    size = "small",
+    size = "medium",
     tags = ["team:rllib", "quick_train"],
     args = [
         "--env", "FrozenLake-v1",
@@ -1324,7 +1324,7 @@ py_test(
 py_test(
     name = "test_dqn_cartpole_v0_no_dueling",
     main = "train.py", srcs = ["train.py"],
-    size = "small",
+    size = "medium",
     tags = ["team:rllib", "quick_train"],
     args = [
         "--env", "CartPole-v0",
@@ -1351,7 +1351,7 @@ py_test(
     name = "test_dqn_cartpole_v0_with_offline_input_and_softq",
     main = "train.py", srcs = ["train.py"],
     tags = ["team:rllib", "quick_train", "external_files"],
-    size = "small",
+    size = "medium",
     # Include the json data file.
     data = ["tests/data/cartpole/small.json"],
     args = [