Skip to content

Commit

Permalink
[Core] Remove grpcio import from usage_lib (ray-project#36542)
Browse files Browse the repository at this point in the history
This is needed for ray-project#35472
  • Loading branch information
pcmoritz committed Jun 19, 2023
1 parent 8290bd1 commit 1529f1d
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions python/ray/_private/usage/usage_lib.py
Original file line number Diff line number Diff line change
Expand Up @@ -60,9 +60,8 @@
import ray
import ray._private.ray_constants as ray_constants
import ray._private.usage.usage_constants as usage_constant
from ray._private import gcs_utils
from ray.experimental.internal_kv import _internal_kv_initialized, _internal_kv_put
from ray.core.generated import usage_pb2
from ray.core.generated import usage_pb2, gcs_pb2

logger = logging.getLogger(__name__)
TagKey = usage_pb2.TagKey
Expand Down Expand Up @@ -496,7 +495,7 @@ def get_total_num_nodes_to_report(gcs_client, timeout=None) -> Optional[int]:
result = gcs_client.get_all_node_info(timeout=timeout)
total_num_nodes = 0
for node_id, node_info in result.items():
if node_info["state"] == gcs_utils.GcsNodeInfo.GcsNodeState.ALIVE:
if node_info["state"] == gcs_pb2.GcsNodeInfo.GcsNodeState.ALIVE:
total_num_nodes += 1
return total_num_nodes
except Exception as e:
Expand Down

0 comments on commit 1529f1d

Please sign in to comment.