skypilot-nightly 1.0.0.dev20241023__py3-none-any.whl → 1.0.0.dev20241025__py3-none-any.whl

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.
Files changed (31) hide show
  1. sky/__init__.py +4 -2
  2. sky/adaptors/azure.py +11 -0
  3. sky/check.py +11 -4
  4. sky/cli.py +24 -16
  5. sky/clouds/azure.py +86 -50
  6. sky/clouds/cloud.py +4 -0
  7. sky/clouds/cloud_registry.py +55 -10
  8. sky/clouds/kubernetes.py +1 -1
  9. sky/clouds/oci.py +1 -1
  10. sky/clouds/service_catalog/azure_catalog.py +15 -0
  11. sky/clouds/service_catalog/kubernetes_catalog.py +7 -1
  12. sky/clouds/utils/azure_utils.py +91 -0
  13. sky/exceptions.py +4 -4
  14. sky/jobs/recovery_strategy.py +3 -3
  15. sky/provision/azure/azure-config-template.json +7 -1
  16. sky/provision/azure/config.py +24 -8
  17. sky/provision/azure/instance.py +251 -137
  18. sky/provision/kubernetes/instance.py +4 -2
  19. sky/provision/provisioner.py +16 -8
  20. sky/resources.py +1 -0
  21. sky/templates/azure-ray.yml.j2 +2 -0
  22. sky/usage/usage_lib.py +3 -2
  23. sky/utils/common_utils.py +3 -2
  24. sky/utils/controller_utils.py +69 -18
  25. {skypilot_nightly-1.0.0.dev20241023.dist-info → skypilot_nightly-1.0.0.dev20241025.dist-info}/METADATA +1 -1
  26. {skypilot_nightly-1.0.0.dev20241023.dist-info → skypilot_nightly-1.0.0.dev20241025.dist-info}/RECORD +30 -30
  27. sky/provision/azure/azure-vm-template.json +0 -301
  28. {skypilot_nightly-1.0.0.dev20241023.dist-info → skypilot_nightly-1.0.0.dev20241025.dist-info}/LICENSE +0 -0
  29. {skypilot_nightly-1.0.0.dev20241023.dist-info → skypilot_nightly-1.0.0.dev20241025.dist-info}/WHEEL +0 -0
  30. {skypilot_nightly-1.0.0.dev20241023.dist-info → skypilot_nightly-1.0.0.dev20241025.dist-info}/entry_points.txt +0 -0
  31. {skypilot_nightly-1.0.0.dev20241023.dist-info → skypilot_nightly-1.0.0.dev20241025.dist-info}/top_level.txt +0 -0
sky/provision/azure/azure-vm-template.json DELETED
@@ -1,301 +0,0 @@
1
- {
2
- "$schema": "https://schema.management.azure.com/schemas/2019-04-01/deploymentTemplate.json#",
3
- "contentVersion": "1.0.0.0",
4
- "parameters": {
5
- "vmName": {
6
- "type": "string",
7
- "metadata": {
8
- "description": "The name of you Virtual Machine."
9
- }
10
- },
11
- "adminUsername": {
12
- "type": "string",
13
- "metadata": {
14
- "description": "Username for the Virtual Machine."
15
- }
16
- },
17
- "publicKey": {
18
- "type": "securestring",
19
- "metadata": {
20
- "description": "SSH Key for the Virtual Machine"
21
- }
22
- },
23
- "imagePublisher": {
24
- "type": "string",
25
- "metadata": {
26
- "description": "The publisher of the VM image"
27
- }
28
- },
29
- "imageOffer": {
30
- "type": "string",
31
- "metadata": {
32
- "description": "The offer of the VM image"
33
- }
34
- },
35
- "imageSku": {
36
- "type": "string",
37
- "metadata": {
38
- "description": "The sku of the VM image"
39
- }
40
- },
41
- "imageVersion": {
42
- "type": "string",
43
- "metadata": {
44
- "description": "The version of the VM image"
45
- }
46
- },
47
- "vmSize": {
48
- "type": "string",
49
- "metadata": {
50
- "description": "The size of the VM"
51
- }
52
- },
53
- "vmTags": {
54
- "type": "object",
55
- "metadata": {
56
- "description": "Tags for the VM"
57
- }
58
- },
59
- "vmCount": {
60
- "type": "int",
61
- "metadata": {
62
- "description": "Number of VMs to deploy"
63
- }
64
- },
65
- "provisionPublicIp": {
66
- "type": "bool",
67
- "defaultValue": true,
68
- "metadata": {
69
- "description": "If true creates a public ip"
70
- }
71
- },
72
- "priority": {
73
- "type": "string",
74
- "defaultValue": "Regular",
75
- "metadata": {
76
- "description": "Specifies the priority for the virtual machine."
77
- }
78
- },
79
- "billingProfile": {
80
- "type": "object",
81
- "defaultValue": {},
82
- "metadata": {
83
- "description": "Specifies the maximum price to pay for Azure Spot VM."
84
- }
85
- },
86
- "osDiskSizeGB": {
87
- "type": "int",
88
- "metadata": {
89
- "description": "OS disk size in GBs."
90
- }
91
- },
92
- "msi": {
93
- "type": "string",
94
- "metadata": {
95
- "description": "Managed service identity resource id."
96
- }
97
- },
98
- "nsg": {
99
- "type": "string",
100
- "metadata": {
101
- "description": "Network security group resource id."
102
- }
103
- },
104
- "subnet": {
105
- "type": "string",
106
- "metadata": {
107
- "descriptions": "Subnet resource id."
108
- }
109
- },
110
- "osDiskTier": {
111
- "type": "string",
112
- "allowedValues": [
113
- "Premium_LRS",
114
- "StandardSSD_LRS",
115
- "Standard_LRS"
116
- ],
117
- "metadata": {
118
- "description": "OS disk tier."
119
- }
120
- },
121
- "cloudInitSetupCommands": {
122
- "type": "string",
123
- "metadata": {
124
- "description": "Base64 encoded cloud-init setup commands."
125
- }
126
- }
127
- },
128
- "variables": {
129
- "location": "[resourceGroup().location]",
130
- "networkInterfaceNamePrivate": "[concat(parameters('vmName'), '-nic')]",
131
- "networkInterfaceNamePublic": "[concat(parameters('vmName'), '-nic-public')]",
132
- "networkInterfaceName": "[if(parameters('provisionPublicIp'), variables('networkInterfaceNamePublic'), variables('networkInterfaceNamePrivate'))]",
133
- "networkIpConfig": "[guid(resourceGroup().id, parameters('vmName'))]",
134
- "publicIpAddressName": "[concat(parameters('vmName'), '-ip')]"
135
- },
136
- "resources": [
137
- {
138
- "type": "Microsoft.Network/networkInterfaces",
139
- "apiVersion": "2020-06-01",
140
- "name": "[concat(variables('networkInterfaceNamePublic'), copyIndex())]",
141
- "location": "[variables('location')]",
142
- "dependsOn": [
143
- "[resourceId('Microsoft.Network/publicIpAddresses/', concat(variables('publicIpAddressName'), copyIndex()))]"
144
- ],
145
- "copy": {
146
- "name": "NICPublicCopy",
147
- "count": "[parameters('vmCount')]"
148
- },
149
- "properties": {
150
- "ipConfigurations": [
151
- {
152
- "name": "[variables('networkIpConfig')]",
153
- "properties": {
154
- "subnet": {
155
- "id": "[parameters('subnet')]"
156
- },
157
- "privateIPAllocationMethod": "Dynamic",
158
- "publicIpAddress": {
159
- "id": "[resourceId('Microsoft.Network/publicIPAddresses', concat(variables('publicIPAddressName'), copyIndex()))]"
160
- }
161
- }
162
- }
163
- ],
164
- "networkSecurityGroup": {
165
- "id": "[parameters('nsg')]"
166
- }
167
- },
168
- "condition": "[parameters('provisionPublicIp')]"
169
- },
170
- {
171
- "type": "Microsoft.Network/networkInterfaces",
172
- "apiVersion": "2020-06-01",
173
- "name": "[concat(variables('networkInterfaceNamePrivate'), copyIndex())]",
174
- "location": "[variables('location')]",
175
- "copy": {
176
- "name": "NICPrivateCopy",
177
- "count": "[parameters('vmCount')]"
178
- },
179
- "properties": {
180
- "ipConfigurations": [
181
- {
182
- "name": "[variables('networkIpConfig')]",
183
- "properties": {
184
- "subnet": {
185
- "id": "[parameters('subnet')]"
186
- },
187
- "privateIPAllocationMethod": "Dynamic"
188
- }
189
- }
190
- ],
191
- "networkSecurityGroup": {
192
- "id": "[parameters('nsg')]"
193
- }
194
- },
195
- "condition": "[not(parameters('provisionPublicIp'))]"
196
- },
197
- {
198
- "type": "Microsoft.Network/publicIpAddresses",
199
- "apiVersion": "2019-02-01",
200
- "name": "[concat(variables('publicIpAddressName'), copyIndex())]",
201
- "location": "[variables('location')]",
202
- "properties": {
203
- "publicIpAllocationMethod": "Static",
204
- "publicIPAddressVersion": "IPv4"
205
- },
206
- "copy": {
207
- "name": "PublicIpCopy",
208
- "count": "[parameters('vmCount')]"
209
- },
210
- "sku": {
211
- "name": "Basic",
212
- "tier": "Regional"
213
- },
214
- "condition": "[parameters('provisionPublicIp')]"
215
- },
216
- {
217
- "type": "Microsoft.Compute/virtualMachines",
218
- "apiVersion": "2019-03-01",
219
- "name": "[concat(parameters('vmName'), copyIndex())]",
220
- "location": "[variables('location')]",
221
- "dependsOn": [
222
- "[resourceId('Microsoft.Network/networkInterfaces/', concat(variables('networkInterfaceName'), copyIndex()))]"
223
- ],
224
- "copy": {
225
- "name": "VmCopy",
226
- "count": "[parameters('vmCount')]"
227
- },
228
- "tags": "[parameters('vmTags')]",
229
- "properties": {
230
- "hardwareProfile": {
231
- "vmSize": "[parameters('vmSize')]"
232
- },
233
- "storageProfile": {
234
- "osDisk": {
235
- "createOption": "fromImage",
236
- "managedDisk": {
237
- "storageAccountType": "[parameters('osDiskTier')]"
238
- },
239
- "diskSizeGB": "[parameters('osDiskSizeGB')]"
240
- },
241
- "imageReference": {
242
- "publisher": "[parameters('imagePublisher')]",
243
- "offer": "[parameters('imageOffer')]",
244
- "sku": "[parameters('imageSku')]",
245
- "version": "[parameters('imageVersion')]"
246
- }
247
- },
248
- "networkProfile": {
249
- "networkInterfaces": [
250
- {
251
- "id": "[resourceId('Microsoft.Network/networkInterfaces', concat(variables('networkInterfaceName'), copyIndex()))]"
252
- }
253
- ]
254
- },
255
- "osProfile": {
256
- "computerName": "[concat(parameters('vmName'), copyIndex())]",
257
- "adminUsername": "[parameters('adminUsername')]",
258
- "adminPassword": "[parameters('publicKey')]",
259
- "linuxConfiguration": {
260
- "disablePasswordAuthentication": true,
261
- "ssh": {
262
- "publicKeys": [
263
- {
264
- "path": "[concat('/home/', parameters('adminUsername'), '/.ssh/authorized_keys')]",
265
- "keyData": "[parameters('publicKey')]"
266
- }
267
- ]
268
- }
269
- },
270
- "customData": "[parameters('cloudInitSetupCommands')]"
271
- },
272
- "priority": "[parameters('priority')]",
273
- "billingProfile": "[parameters('billingProfile')]"
274
- },
275
- "identity": {
276
- "type": "UserAssigned",
277
- "userAssignedIdentities": {
278
- "[parameters('msi')]": {
279
- }
280
- }
281
- }
282
- }
283
- ],
284
- "outputs": {
285
- "publicIp": {
286
- "type": "array",
287
- "copy": {
288
- "count": "[parameters('vmCount')]",
289
- "input": "[reference(concat(variables('publicIpAddressName'), copyIndex())).ipAddress]"
290
- },
291
- "condition": "[parameters('provisionPublicIp')]"
292
- },
293
- "privateIp": {
294
- "type": "array",
295
- "copy": {
296
- "count": "[parameters('vmCount')]",
297
- "input": "[reference(concat(variables('networkInterfaceName'), copyIndex())).ipConfigurations[0].properties.privateIPAddress]"
298
- }
299
- }
300
- }
301
- }