Skip to content

Commit

Permalink
fix configurations docs for skypilot (#2017)
Browse files Browse the repository at this point in the history
  • Loading branch information
safoinme authored Nov 3, 2023
1 parent c32ec57 commit 3ea1811
Showing 1 changed file with 7 additions and 9 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -266,8 +266,7 @@ The following code snippets show how to configure the orchestrator settings for
**Code Example:**

```python
from zenml.integrations.skypilot.flavors.skypilot_orchestrator_flavor import SkypilotAWSOrchestratorSettings

from zenml.integrations.skypilot.flavors.skypilot_orchestrator_aws_vm_flavor import SkypilotAWSOrchestratorSettings

skypilot_settings = SkypilotAWSOrchestratorSettings(
cpus="2",
Expand All @@ -276,7 +275,7 @@ skypilot_settings = SkypilotAWSOrchestratorSettings(
accelerator_args={"tpu_vm": True, "runtime_version": "tpu-vm-base"},
use_spot=True,
spot_recovery="recovery_strategy",
region="us-west1",
region="us-west-1",
zone="us-west1-a",
image_id="ami-1234567890abcdef0",
disk_size=100,
Expand All @@ -303,7 +302,7 @@ skypilot_settings = SkypilotAWSOrchestratorSettings(
**Code Example:**

```python
from zenml.integrations.skypilot.flavors.skypilot_orchestrator_flavor import SkypilotGCPOrchestratorSettings
from zenml.integrations.skypilot.flavors.skypilot_orchestrator_gcp_vm_flavor import SkypilotGCPOrchestratorSettings


skypilot_settings = SkypilotGCPOrchestratorSettings(
Expand All @@ -315,7 +314,7 @@ skypilot_settings = SkypilotGCPOrchestratorSettings(
spot_recovery="recovery_strategy",
region="us-west1",
zone="us-west1-a",
image_id="ami-1234567890abcdef0",
image_id="ubuntu-pro-2004-focal-v20231101",
disk_size=100,
disk_tier="high",
cluster_name="my_cluster",
Expand All @@ -340,7 +339,7 @@ skypilot_settings = SkypilotGCPOrchestratorSettings(
**Code Example:**

```python
from zenml.integrations.skypilot.flavors.skypilot_orchestrator_flavor import SkypilotAzureOrchestratorSettings
from zenml.integrations.skypilot.flavors.skypilot_orchestrator_azure_vm_flavor import SkypilotAzureOrchestratorSettings


skypilot_settings = SkypilotAzureOrchestratorSettings(
Expand All @@ -350,9 +349,8 @@ skypilot_settings = SkypilotAzureOrchestratorSettings(
accelerator_args={"tpu_vm": True, "runtime_version": "tpu-vm-base"},
use_spot=True,
spot_recovery="recovery_strategy",
region="us-west1",
zone="us-west1-a",
image_id="ami-1234567890abcdef0",
region="West Europe",
image_id="Canonical:0001-com-ubuntu-server-jammy:22_04-lts-gen2:latest",
disk_size=100,
disk_tier="high",
cluster_name="my_cluster",
Expand Down

0 comments on commit 3ea1811

Please sign in to comment.