mirror of
https://github.com/microsoft/SkillOpt.git
synced 2026-07-07 00:15:39 +08:00
14 lines
489 B
Python
14 lines
489 B
Python
from skillopt.datasets.base import SplitDataLoader
|
|
from skillopt.envs._template.env_template import TemplateBenchmarkAdapter
|
|
from skillopt.envs._template.loader_template import TemplateBenchmarkDataLoader
|
|
|
|
|
|
def test_template_adapter_is_concrete():
|
|
adapter = TemplateBenchmarkAdapter()
|
|
assert adapter.get_task_types() == ["your_benchmark"]
|
|
|
|
|
|
def test_template_loader_uses_split_dataloader():
|
|
loader = TemplateBenchmarkDataLoader()
|
|
assert isinstance(loader, SplitDataLoader)
|