Bump google-github-actions/auth from 2 to 3

.github/workflows/gcp.yaml

@@ -36,7 +36,7 @@ jobs:
       uses: actions/checkout@v5
 
     - name: Auth with GCP
-      uses: google-github-actions/auth@v2
+      uses: google-github-actions/auth@v3
       with:
         credentials_json: ${{ secrets.GCLOUD_SERVICE_KEY }}
 

src/ml_ops/link_best_model.py

@@ -1,9 +1,8 @@
 # Takes a sweep ID, finds the best run, and links its model artifact to a model registry.
 
 import typer
-from loguru import logger
-
 import wandb
+from loguru import logger
 
 app = typer.Typer()
 

src/ml_ops/link_model.py

@@ -1,7 +1,6 @@
 import os
 
 import typer
-
 import wandb
 
 

src/ml_ops/promote_model.py

@@ -2,9 +2,8 @@
 import operator
 
 import typer
-from dotenv import load_dotenv
-
 import wandb
+from dotenv import load_dotenv
 
 # Initialize Typer app
 app = typer.Typer()

src/ml_ops/train.py

@@ -5,6 +5,7 @@
 import hydra
 import matplotlib.pyplot as plt
 import torch
+import wandb
 from dotenv import load_dotenv
 from hydra.utils import instantiate
 from loguru import logger
@@ -13,7 +14,6 @@
 from torch.utils.data import DataLoader
 from torchvision.utils import make_grid
 
-import wandb
 from ml_ops.data import corrupt_mnist
 from ml_ops.device import DEVICE
 from ml_ops.model import MyAwesomeModel

src/ml_ops/train_cli.py

@@ -8,6 +8,7 @@
 import matplotlib.pyplot as plt
 import torch
 import typer
+import wandb
 from dotenv import load_dotenv
 from hydra.utils import instantiate
 from loguru import logger
@@ -16,7 +17,6 @@
 from torch.utils.data import DataLoader
 from torchvision.utils import make_grid
 
-import wandb
 from ml_ops.data import corrupt_mnist
 from ml_ops.device import DEVICE
 from ml_ops.model import MyAwesomeModel

tests/performancetests/test_model.py

@@ -11,8 +11,8 @@
 import time
 
 import torch
-
 import wandb
+
 from ml_ops.model import MyAwesomeModel