fixed test url

Genentech · Jul 24, 2024 · dedfa8c · dedfa8c
1 parent 096ca2a
commit dedfa8c
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/tests/test_models.py b/tests/test_models.py
@@ -1,5 +1,6 @@
 import torch
 
+import wandb
 from grelu.model.models import (
     BorzoiModel,
     BorzoiPretrainedModel,
@@ -11,11 +12,18 @@
     EnformerModel,
     EnformerPretrainedModel,
 )
+from grelu.resources import DEFAULT_WANDB_HOST
 from grelu.sequence.format import convert_input_type
 
 inputs = convert_input_type(["A" * 128], "one_hot")
 
 
+try:
+    wandb.login(host=DEFAULT_WANDB_HOST, anonymous="never", timeout=0)
+except wandb.errors.UsageError:  # login anonymously if not logged in already
+    wandb.login(host=DEFAULT_WANDB_HOST, relogin=True, anonymous="must", timeout=0)
+
+
 # Test a fully convolutional model with residual connections and autocropping
 def test_conv_model():
     model = ConvModel(