Skip to content

Commit

Permalink
Fixed wandb login problem in Quickstart (#3233)
Browse files Browse the repository at this point in the history
* Added wandb env variable

* Disable WANDB in training configurations for AWS, Azure, GCP.- Set WANDB_DISABLED to true in environment variables
  • Loading branch information
htahir1 authored Dec 2, 2024
1 parent 766fb69 commit 634d034
Show file tree
Hide file tree
Showing 5 changed files with 760 additions and 746 deletions.
3 changes: 3 additions & 0 deletions examples/quickstart/configs/training_aws.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,9 @@ settings:
parent_image: "715803424590.dkr.ecr.eu-central-1.amazonaws.com/zenml-public-pipelines:quickstart-0.70.0-py3.11-aws"
skip_build: True # If you switch this to False remove the parent_image
requirements: requirements.txt
environment:
WANDB_DISABLED: "true"

orchestrator.sagemaker:
instance_type: ml.m5.4xlarge

Expand Down
2 changes: 2 additions & 0 deletions examples/quickstart/configs/training_azure.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,8 @@ settings:
parent_image: "zenmldocker/zenml-public-pipelines:quickstart-0.70.0-py3.11-azure"
skip_build: True
requirements: requirements.txt
environment:
WANDB_DISABLED: "true"
# Uncomment the following lines to specify the accelerator for your azureml orchestrator
# orchestrator.azureml:
# mode: "compute-instance"
Expand Down
2 changes: 2 additions & 0 deletions examples/quickstart/configs/training_default.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,8 @@
settings:
docker:
requirements: requirements.txt
environment:
WANDB_DISABLED: "true"

# Model Control Plane configuration
model:
Expand Down
2 changes: 2 additions & 0 deletions examples/quickstart/configs/training_gcp.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,8 @@ settings:
parent_image: "zenmldocker/zenml-public-pipelines:quickstart-0.70.0-py3.11-gcp"
skip_build: True
requirements: requirements.txt
environment:
WANDB_DISABLED: "true"
# Uncomment the following two lines to specify the accelerator for your vertex orchestrator
# orchestrator.vertex:
# node_selector_constraint: ["cloud.google.com/gke-accelerator", "NVIDIA_TESLA_P4"]
Expand Down
Loading

0 comments on commit 634d034

Please sign in to comment.