fix

Signed-off-by: sven1977 <[email protected]>
ray-project · May 2, 2024 · 311ea3f · 311ea3f
1 parent d08c0e5
commit 311ea3f
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/rllib/BUILD b/rllib/BUILD
@@ -2279,11 +2279,10 @@ py_test(
 )
 
 py_test(
-    name = "examples/envs/env_rendering_and_recording.py",
-    main = "examples/envs/env_rendering_and_recording.py",
+    name = "examples/envs/env_rendering_and_recording",
+    srcs = ["examples/envs/env_rendering_and_recording.py"],
     tags = ["team:rllib", "exclusive", "examples"],
     size = "small",
-    srcs = ["examples/envs/env_rendering_and_recording.py"],
     args = ["--enable-new-api-stack", "--env=CartPole-v1", "--stop-iters=3"]
 )