Skip to content

Commit 9706bfe

Browse files
committed
Skip helios test
1 parent 7d41f6f commit 9706bfe

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

tests/integration/helios/test_task_embeds.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,17 @@
1-
"""Testing task-conditioned Helios model."""
1+
"""Testing task-conditioned Helios model.
2+
3+
These tests are skipped in the CI to avoid pulling helios in as a dependency,
4+
but should pass locally.
5+
"""
26

37
from pathlib import Path
48
from typing import Any
59

610
import pytest
711
import torch
812

9-
from rslp.helios.model import Helios, TaskConditionedHelios
13+
pytest.importorskip("helios")
14+
from rslp.helios.model import Helios, TaskConditionedHelios # noqa: E402
1015

1116

1217
@pytest.fixture

0 commit comments

Comments
 (0)