We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7d41f6f commit 9706bfeCopy full SHA for 9706bfe
tests/integration/helios/test_task_embeds.py
@@ -1,12 +1,17 @@
1
-"""Testing task-conditioned Helios model."""
+"""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
+"""
6
7
from pathlib import Path
8
from typing import Any
9
10
import pytest
11
import torch
12
-from rslp.helios.model import Helios, TaskConditionedHelios
13
+pytest.importorskip("helios")
14
+from rslp.helios.model import Helios, TaskConditionedHelios # noqa: E402
15
16
17
@pytest.fixture
0 commit comments