File size: 350 Bytes
eb9ae1f
386fb31
eb9ae1f
386fb31
eb9ae1f
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16

from transformers import GPT2LMHeadModel, GPT2TokenizerFast

from hexviz.models import get_zymctrl


def test_get_zymctrl():
    result = get_zymctrl()

    assert result is not None
    assert isinstance(result, tuple)

    tokenizer, model = result

    assert isinstance(tokenizer, GPT2TokenizerFast)
    assert isinstance(model, GPT2LMHeadModel)