[ci] apply isort to release test directory, part 1

excluding `*_tests` directories for now to reduce the impact

Signed-off-by: Lonnie Liu <lonnie@anyscale.com>

Author: aslonnie

pyproject.toml

@@ -65,7 +65,7 @@ afterray = ["psutil", "setproctitle"]
 "python/ray/__init__.py" = ["I"]
 "python/ray/dag/__init__.py" = ["I"]
 "python/ray/air/__init__.py" = ["I"]
-"release/*" = ["I"]
+"release/*_tests/*.py" = ["I"]
 
 # TODO(matthewdeng): Remove this line
 "python/ray/tune/__init__.py" = ["I"]

release/benchmark-worker-startup/benchmark_worker_startup.py

@@ -40,17 +40,18 @@
 measurements.
 """
 
-from collections import defaultdict
-from dataclasses import dataclass
-from ray._private.test_utils import safe_write_to_results_json
-from ray.job_submission import JobSubmissionClient, JobStatus
 import argparse
 import asyncio
 import random
-import ray
 import statistics
 import subprocess
 import sys
+from collections import defaultdict
+from dataclasses import dataclass
+
+import ray
+from ray._private.test_utils import safe_write_to_results_json
+from ray.job_submission import JobStatus, JobSubmissionClient
 
 
 def main(

release/benchmark-worker-startup/test_single_configuration.py

@@ -5,10 +5,11 @@
 """
 
 import argparse
-import ray
 import sys
 import time
 
+import ray
+
 
 @ray.remote
 class Actor:

release/benchmarks/distributed/test_many_actors.py

@@ -1,11 +1,12 @@
 import os
-import ray
-import ray._common.test_utils
-import ray._private.test_utils as test_utils
 import time
-import tqdm
 
+import tqdm
 from many_nodes_tests.dashboard_test import DashboardTestAtScale
+
+import ray
+import ray._common.test_utils
+import ray._private.test_utils as test_utils
 from ray._private.state_api_test_utils import summarize_worker_startup_time
 
 is_smoke_test = True

release/benchmarks/distributed/test_many_pgs.py

@@ -1,12 +1,14 @@
 import os
+import time
+
+import tqdm
+from many_nodes_tests.dashboard_test import DashboardTestAtScale
+
 import ray
 import ray._common.test_utils
 import ray._private.test_utils as test_utils
 from ray.util.placement_group import placement_group, remove_placement_group
 from ray.util.scheduling_strategies import PlacementGroupSchedulingStrategy
-from many_nodes_tests.dashboard_test import DashboardTestAtScale
-import time
-import tqdm
 
 is_smoke_test = True
 if "SMOKE_TEST" in os.environ:

release/benchmarks/distributed/test_many_tasks.py

@@ -1,17 +1,18 @@
+import time
+
 import click
+import tqdm
+from many_nodes_tests.dashboard_test import DashboardTestAtScale
+
 import ray
 import ray._common.test_utils
 import ray._private.test_utils as test_utils
-import time
-import tqdm
-
-from ray.util.state import summarize_tasks
-from many_nodes_tests.dashboard_test import DashboardTestAtScale
 from ray._private.state_api_test_utils import (
     StateAPICallSpec,
     periodic_invoke_state_apis_with_actor,
     summarize_worker_startup_time,
 )
+from ray.util.state import summarize_tasks
 
 sleep_time = 300
 

release/benchmarks/distributed/test_scheduling.py

@@ -1,9 +1,10 @@
-import ray
 import argparse
-from time import time, sleep
 from math import floor
-from ray._private.test_utils import safe_write_to_results_json
+from time import sleep, time
+
+import ray
 import ray._private.test_utils as test_utils
+from ray._private.test_utils import safe_write_to_results_json
 
 
 @ray.remote

release/benchmarks/object_store/test_large_objects.py

@@ -1,12 +1,11 @@
-import numpy as np
-
-import ray
-
 import json
 import os
 from time import perf_counter
+
+import numpy as np
 from tqdm import tqdm
 
+import ray
 
 NUM_NODES = 9
 OBJECT_SIZE = 2**32

release/benchmarks/object_store/test_object_store.py

@@ -1,13 +1,13 @@
-import numpy as np
-
-import ray
-import ray.autoscaler.sdk
-
 import json
 import os
 from time import perf_counter
+
+import numpy as np
 from tqdm import tqdm
 
+import ray
+import ray.autoscaler.sdk
+
 NUM_NODES = 50
 OBJECT_SIZE = 2**30
 

release/benchmarks/object_store/test_small_objects.py

@@ -1,8 +1,10 @@
-import ray
-import numpy as np
-import time
-import os
 import json
+import os
+import time
+
+import numpy as np
+
+import ray
 
 
 def test_small_objects_many_to_one():