chore: add simple test to fix github action
This commit is contained in:
parent
365e0fb931
commit
9ab864fbdf
13
tests/test_models.py
Normal file
13
tests/test_models.py
Normal 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)
|
Loading…
Reference in New Issue
Block a user