ray-embedding 0.13.9__tar.gz → 0.13.10__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of ray-embedding might be problematic. Click here for more details.

@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: ray-embedding
3
- Version: 0.13.9
3
+ Version: 0.13.10
4
4
  Summary: Deploy SentenceTransformers embedding models to a ray cluster
5
5
  Author: Crispin Almodovar
6
6
  Author-email:
@@ -76,19 +76,22 @@ class NodeHealthTracker:
76
76
  target_node_id = next(node["NodeID"] for node in ray.nodes() if node["Alive"] and node["NodeManagerAddress"] == node_ip)
77
77
  assert target_node_id, f"No node found with IP {node_ip}"
78
78
  prefixes = tuple(f"SERVE_REPLICA::{self.app_name}#{d}" for d in self.tracked_model_deployments)
79
- self.logger.info(f"Checking actors with prefixes: {prefixes} in node IP {node_ip}, ID {target_node_id}")
80
79
 
81
- for actor in list_actors(detail=False, filters=[("node_id", "=", target_node_id)]):
82
- self.logger.info(f"Checking actor: {actor}")
83
- if actor.state in ["DEPENDENCIES_UNREADY", 'PENDING_CREATION', 'ALIVE', 'RESTARTING']:
84
- for prefix in prefixes:
85
- if actor.name.startswith(prefix):
86
- self.logger.info(f"Found a replica {actor.name} of "
87
- f"tracked_deployments={self.tracked_model_deployments} "
88
- f"running in node IP {node_ip}, node ID {target_node_id}.")
89
- return True
80
+ actors_list = list_actors(detail=True, filters=[("node_id", "=", target_node_id), ("state", "!=", "DEAD")])
81
+ if actors_list:
82
+ self.logger.info(f"Checking non-dead actors with prefixes: {prefixes} in node IP {node_ip}, ID {target_node_id}")
83
+ for actor in actors_list:
84
+ self.logger.info(f"Checking actor: {actor}")
85
+ if actor.state in ["DEPENDENCIES_UNREADY", 'PENDING_CREATION', 'ALIVE', 'RESTARTING']:
86
+ for prefix in prefixes:
87
+ if actor.name.startswith(prefix):
88
+ self.logger.info(f"Found a replica {actor.name} of "
89
+ f"tracked_deployments={self.tracked_model_deployments} "
90
+ f"running in node IP {node_ip}, node ID {target_node_id}.")
91
+ return True
92
+
90
93
  self.logger.info(f"No replicas of tracked deployments={self.tracked_model_deployments} running on node: {node_ip}.")
91
- return False
92
94
  except Exception as e:
93
95
  self.logger.error(f"An error occurred while checking replicas on node {node_ip}: {e}")
94
- return False
96
+
97
+ return False
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: ray-embedding
3
- Version: 0.13.9
3
+ Version: 0.13.10
4
4
  Summary: Deploy SentenceTransformers embedding models to a ray cluster
5
5
  Author: Crispin Almodovar
6
6
  Author-email:
@@ -1,6 +1,6 @@
1
1
  [metadata]
2
2
  name = ray-embedding
3
- version = 0.13.9
3
+ version = 0.13.10
4
4
  author = Crispin Almodovar
5
5
  author_email =
6
6
  description = Deploy SentenceTransformers embedding models to a ray cluster