From 75b419279f175e285cf87a2a64b0a8e911bf8ada Mon Sep 17 00:00:00 2001 From: Caroline Date: Mon, 9 Sep 2024 18:04:13 -0400 Subject: [PATCH] Add more to launched properties --- runhouse/resources/hardware/on_demand_cluster.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/runhouse/resources/hardware/on_demand_cluster.py b/runhouse/resources/hardware/on_demand_cluster.py index 6c8253864..ee6b81115 100644 --- a/runhouse/resources/hardware/on_demand_cluster.py +++ b/runhouse/resources/hardware/on_demand_cluster.py @@ -403,12 +403,21 @@ def _populate_connection_from_status_dict(self, cluster_dict: Dict[str, Any]): instance_type = launched_resource.instance_type region = launched_resource.region cost_per_hr = launched_resource.get_cost(60 * 60) + disk_size = launched_resource.get("disk_size") + num_cpus = launched_resource.cpus + self.launched_properties = { "cloud": cloud, "instance_type": instance_type, "region": region, "cost_per_hour": str(cost_per_hr), + "disk_size": disk_size, + "num_cpus": num_cpus, } + if launched_resource.accelerators: + self.launched_properties[ + "accelerators" + ] = launched_resource.accelerators if handle.ssh_user: self.launched_properties["ssh_user"] = handle.ssh_user if handle.docker_user: