chore: add simple test to fix github action

This commit is contained in:
Alexander Engelsberger 2022-10-26 12:57:45 +02:00
parent 365e0fb931
commit 9ab864fbdf
No known key found for this signature in database

13
tests/test_models.py Normal file
View File

@ -0,0 +1,13 @@
"""prototorch.models test suite."""
import prototorch as pt
from prototorch.models.library import GLVQ
def test_glvq_model_build():
hparams = GLVQ.HyperParameters(
distribution=dict(num_classes=2, per_class=1),
component_initializer=pt.initializers.RNCI(2),
)
model = GLVQ(hparams=hparams)