17 lines
397 B
Python
17 lines
397 B
Python
import pytest
|
|
|
|
from tcalc.dataset import TCalcDataset
|
|
from tcalc.model import TCalcPredictor
|
|
from tcalc.utils import read_yaml
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def dataset():
|
|
return TCalcDataset(read_yaml("config/dataset.yaml"))
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def tcalc_predictor():
|
|
p = TCalcPredictor(read_yaml("config/dataset.yaml"))
|
|
p.load_models("default_model")
|
|
return p |