Skip to content

Commit dedfa8c

Browse files
committed
fixed test url
1 parent 096ca2a commit dedfa8c

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

tests/test_models.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
import torch
22

3+
import wandb
34
from grelu.model.models import (
45
BorzoiModel,
56
BorzoiPretrainedModel,
@@ -11,11 +12,18 @@
1112
EnformerModel,
1213
EnformerPretrainedModel,
1314
)
15+
from grelu.resources import DEFAULT_WANDB_HOST
1416
from grelu.sequence.format import convert_input_type
1517

1618
inputs = convert_input_type(["A" * 128], "one_hot")
1719

1820

21+
try:
22+
wandb.login(host=DEFAULT_WANDB_HOST, anonymous="never", timeout=0)
23+
except wandb.errors.UsageError: # login anonymously if not logged in already
24+
wandb.login(host=DEFAULT_WANDB_HOST, relogin=True, anonymous="must", timeout=0)
25+
26+
1927
# Test a fully convolutional model with residual connections and autocropping
2028
def test_conv_model():
2129
model = ConvModel(

0 commit comments

Comments
 (0)