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