Python set_datatype 예제들

예제 #1

파일: test_settings.py 프로젝트: 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")

예제 #2

파일: conftest.py 프로젝트: stnkl/probflow

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

예제 #3

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