Files
publicmatt 6eed08d1ba fix project_slug error.
replace ml_pipeline with project_name or module_name from config
2024-04-06 15:48:29 -07:00

7 lines
172 B
Python

from {{cookiecutter.module_name}} import config
from {{cookiecutter.module_name}}.model.cnn import VGG11
def test_in_channels():
assert config.model.name == 'vgg11'