Skip to content

Commit

Permalink
print cloud info in cluster status (#1209)
Browse files Browse the repository at this point in the history
Co-authored-by: Alexandra Belousov <sashabelousovrh@Alexandras-MacBook-Pro.local>
  • Loading branch information
2 people authored and Alexandra Belousov committed Sep 17, 2024
1 parent 4a44573 commit 16bd237
Showing 1 changed file with 20 additions and 0 deletions.
20 changes: 20 additions & 0 deletions runhouse/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -455,6 +455,26 @@ def _get_resource_link_in_den_ui(cluster_name: str, api_server_url: str):
return link_to_den_dashboard


def _print_cloud_properties(cluster_config: dict):
cloud_properties = cluster_config.get("launched_properties", None)
if not cloud_properties:
return
cloud = cloud_properties.get("cloud")
instance_type = cloud_properties.get("instance_type")
region = cloud_properties.get("region")
cost_per_hour = cloud_properties.get("cost_per_hour")

has_cuda = cluster_config.get("has_cuda", False)
cost_emoji = "💰" if has_cuda else "💸"

num_of_instances = len(cluster_config.get("ips"))
num_of_instances_str = f"{num_of_instances}x " if num_of_instances > 1 else ""

print(
f"🤖 {num_of_instances_str}{cloud} {instance_type} cluster | 🌍 {region} | {cost_emoji} ${cost_per_hour}/hr"
)


def _print_status(status_data: dict, current_cluster: Cluster) -> None:
"""Prints the status of the cluster to the console"""
cluster_config = status_data.get("cluster_config")
Expand Down

0 comments on commit 16bd237

Please sign in to comment.