fix lint error

Signed-off-by: 黑驰 <[email protected]>

Author: wumuzi520

python/ray/data/tests/conftest.py

@@ -1,13 +1,16 @@
 import copy
+import logging
 import os
 import posixpath
 import time
 from collections import defaultdict
+from contextlib import contextmanager
 
 import numpy as np
 import pandas as pd
 import pyarrow as pa
 import pytest
+import requests
 
 import ray
 import ray.util.state
@@ -17,6 +20,7 @@
 from ray._private.utils import _get_pyarrow_version
 from ray.air.constants import TENSOR_COLUMN_NAME
 from ray.air.util.tensor_extensions.arrow import ArrowTensorArray
+from ray.cluster_utils import Cluster, cluster_not_supported
 from ray.data import Schema
 from ray.data.block import BlockExecStats, BlockMetadata
 from ray.data.tests.mock_server import *  # noqa
@@ -25,7 +29,13 @@
 # Trigger pytest hook to automatically zip test cluster logs to archive dir on failure
 from ray.tests.conftest import *  # noqa
 from ray.tests.conftest import pytest_runtest_makereport  # noqa
-from ray.tests.conftest import _ray_start, wait_for_condition
+from ray.tests.conftest import (
+    _ray_start,
+    get_default_fixture_ray_kwargs,
+    wait_for_condition,
+)
+
+logger = logging.getLogger(__name__)
 
 
 @pytest.fixture(scope="module")

python/ray/data/tests/test_strict_in_virtual_cluster.py

@@ -1,8 +1,10 @@
 import logging
-import requests
 import sys
-import time
+import tempfile
+from pathlib import Path
+
 import pytest
+import requests
 
 import ray
 from ray._private.runtime_env.working_dir import upload_working_dir_if_needed
@@ -30,10 +32,10 @@ def ready(self):
     def unready(self):
         self._ready = False
         self._barrier = False
-    
+
     def set_barrier(self):
         self._barrier = True
-    
+
     def get_barrier(self):
         return self._barrier
 
@@ -82,7 +84,7 @@ def get_virtual_cluster_nodes(webui_url, virtual_cluster_id):
 
 def get_job_actors(webui_url, job_id=None):
     resp = requests.get(
-        webui_url + f"/logical/actors",
+        webui_url + "/logical/actors",
         timeout=10,
     )
     resp.raise_for_status()
@@ -174,7 +176,10 @@ def test_auto_parallelism(create_virtual_cluster):
 
 data_size = {data_size}
 signal_actor_name = "{signal_actor_name}"
-print("Job is running, data_size: ", data_size, "signal_actor_name: ", signal_actor_name)
+print("Job is running, data_size: ",
+       data_size,
+       "signal_actor_name: ",
+       signal_actor_name)
 ray.init(address="auto")
 signal_actor = ray.get_actor(signal_actor_name, namespace="storage")
 ray.get(signal_actor.ready.remote())
@@ -220,7 +225,8 @@ def estimate_inmemory_data_size(self):
                 # retrieve job data and check
                 res = ray.get(signal_actor.data.remote())
                 print(
-                    f"Driver detected parallelism: {res}, expect[{i}]: {expected_parallelism[i]}"
+                    f"Driver detected parallelism: {res},"
+                    f" expect[{i}]: {expected_parallelism[i]}"
                 )
                 assert res == expected_parallelism[i]