[RLlib] Remove RLlib CLI files, remove other files not needed anymore. (

#49488)
ray-project · Dec 30, 2024 · 8e32c0a · 8e32c0a
1 parent eca56fa
commit 8e32c0a
Show file tree

Hide file tree

Showing 12 changed files with 70 additions and 2,280 deletions.
diff --git a/.buildkite/rllib.rayci.yml b/.buildkite/rllib.rayci.yml
@@ -123,15 +123,15 @@ steps:
       - bazel run //ci/ray_ci:test_in_docker -- //rllib/... rllib --only-tags torch_2.x_only_benchmark
     depends_on: rllibbuild
 
-  - label: ":brain: rllib: memory leak pytorch tests"
-    tags: rllib
-    instance_type: medium
-    commands:
-      - bazel run //ci/ray_ci:test_in_docker -- //rllib/... rllib
-        --only-tags memory_leak_tests
-        --except-tags flaky
-        --test-arg --framework=torch
-    depends_on: rllibbuild
+  # - label: ":brain: rllib: memory leak pytorch tests"
+  #  tags: rllib
+  #  instance_type: medium
+  #  commands:
+  #    - bazel run //ci/ray_ci:test_in_docker -- //rllib/... rllib
+  #      --only-tags memory_leak_tests
+  #      --except-tags flaky
+  #      --test-arg --framework=torch
+  #  depends_on: rllibbuild
 
   - label: ":brain: rllib: doc tests"
     tags:
@@ -266,18 +266,3 @@ steps:
         --skip-ray-installation # reuse the same docker image as the previous run
     depends_on: rllibbuild
     soft_fail: true
-
-  - label: ":brain: rllib: flaky tests (memory leak)"
-    key: rllib_flaky_tests_03
-    tags:
-      - rllib
-      - skip-on-premerge
-    instance_type: medium
-    commands:
-      - bazel run //ci/ray_ci:test_in_docker -- //rllib/... rllib
-        --run-flaky-tests
-        --only-tags memory_leak_tests
-        --except-tags flaky,gpu,multi_gpu
-        --test-arg --framework=tf2
-    depends_on: rllibbuild
-    soft_fail: true
diff --git a/doc/source/rllib/package_ref/utils.rst b/doc/source/rllib/package_ref/utils.rst
@@ -255,8 +255,4 @@ Other utilities
    :toctree: doc/
 
    utils.tensor_dtype.get_np_dtype
-   common.CLIArguments
-   common.FrameworkEnum
-   common.SupportedFileType
    core.rl_module.validate_module_id
-   train.load_experiments_from_file
diff --git a/rllib/BUILD b/rllib/BUILD
@@ -1051,44 +1051,6 @@ py_test(
     srcs = ["algorithms/sac/tests/test_sac.py"]
 )
 
-# --------------------------------------------------------------------
-# Memory leak tests
-#
-# Tag: memory_leak_tests
-# --------------------------------------------------------------------
-
-# @OldAPIStack
-py_test(
-    name = "test_memory_leak_appo_old_api_stack",
-    tags = ["team:rllib", "memory_leak_tests"],
-    main = "utils/tests/run_memory_leak_tests.py",
-    size = "large",
-    srcs = ["utils/tests/run_memory_leak_tests.py"],
-    data = ["tuned_examples/appo/memory-leak-test-appo.yaml"],
-    args = ["--dir=tuned_examples/appo"]
-)
-# @OldAPIStack
-py_test(
-    name = "test_memory_leak_impala_old_api_stack",
-    tags = ["team:rllib", "memory_leak_tests"],
-    main = "utils/tests/run_memory_leak_tests.py",
-    size = "large",
-    srcs = ["utils/tests/run_memory_leak_tests.py"],
-    data = ["tuned_examples/impala/memory-leak-test-impala.yaml"],
-    args = ["--dir=tuned_examples/impala"]
-)
-# @OldAPIStack
-py_test(
-    name = "test_memory_leak_ppo_old_api_stack",
-    tags = ["team:rllib", "memory_leak_tests"],
-    main = "utils/tests/run_memory_leak_tests.py",
-    size = "large",
-    srcs = ["utils/tests/run_memory_leak_tests.py"],
-    data = ["tuned_examples/ppo/memory-leak-test-ppo.yaml"],
-    args = ["--dir=tuned_examples/ppo"]
-)
-
-
 # --------------------------------------------------------------------
 # Connector(V1) tests
 # rllib/connector/