Fix linter

Signed-off-by: jeffreywang <jeffreywang@anyscale.com>

Author: jeffreywang-anyscale

python/ray/serve/_private/deployment_state.py

@@ -37,7 +37,6 @@
     RunningReplicaInfo,
 )
 from ray.serve._private.config import DeploymentConfig
-from ray.serve.config import GangRuntimeFailurePolicy
 from ray.serve._private.constants import (
     DEFAULT_LATENCY_BUCKET_MS,
     MAX_PER_REPLICA_RETRY_COUNT,
@@ -72,6 +71,7 @@
     msgpack_serialize,
 )
 from ray.serve._private.version import DeploymentVersion
+from ray.serve.config import GangRuntimeFailurePolicy
 from ray.serve.generated.serve_pb2 import DeploymentLanguage
 from ray.serve.schema import (
     DeploymentDetails,

python/ray/serve/_private/replica.py

@@ -48,6 +48,7 @@
 from ray.serve._private.common import (
     RUNNING_REQUESTS_KEY,
     DeploymentID,
+    GangContext,
     ReplicaID,
     ReplicaMetricReport,
     ReplicaQueueLengthInfo,

python/ray/serve/config.py

@@ -800,7 +800,10 @@ class GangRuntimeFailurePolicy(str, Enum):
     """Tear down and restart entire gang atomically when any replica fails."""
 
     RESTART_REPLICA = "RESTART_REPLICA"
-    """Tear down and restart individual replica when it fails. Other replicas in the gang will continue running."""
+    """
+    Tear down and restart individual replica when it fails.
+    Other replicas in the gang will continue running.
+    """
 
 
 @PublicAPI(stability="alpha")

python/ray/serve/tests/test_deployment_scheduler.py

@@ -5,7 +5,6 @@
 
 import ray
 from ray import serve
-from ray.serve.config import GangSchedulingConfig, GangPlacementStrategy
 from ray._common.test_utils import wait_for_condition
 from ray._raylet import GcsClient
 from ray.serve._private import default_impl
@@ -17,6 +16,7 @@
 )
 from ray.serve._private.test_utils import check_apps_running, get_node_id
 from ray.serve._private.utils import get_head_node_id
+from ray.serve.config import GangPlacementStrategy, GangSchedulingConfig
 from ray.tests.conftest import *  # noqa