Exemplos de set_datatype em Python

Exemplo n.º 1

Arquivo: test_settings.py Projeto: stnkl/probflow

def test_datatype():
    """Tests get and set_datatype"""

    assert isinstance(pf.get_datatype(), torch.dtype)
    assert pf.get_datatype() == torch.float32

    pf.set_datatype(torch.float64)
    assert isinstance(pf.get_datatype(), torch.dtype)
    assert pf.get_datatype() == torch.float64
    pf.set_datatype(torch.float32)

    with pytest.raises(TypeError):
        pf.set_datatype("lala")

Exemplo n.º 2

Arquivo: conftest.py Projeto: stnkl/probflow

def pytest_runtest_setup(item):
    pf.set_backend("tensorflow")
    pf.set_datatype(None)

Exemplo n.º 3

def pytest_runtest_setup(item):
    pf.set_backend("pytorch")
    pf.set_datatype(None)