Python update_registered_buffersの例

コード例 #1

def test_update_registered_buffers():
    foo = Foo()
    with pytest.raises(ValueError):
        update_registered_buffers(foo, "conv1", ["qweight"], {})

コード例 #2

ファイル: test_models.py プロジェクト: tuzm24/CompressAI

def test_update_registered_buffers():
    foo = Foo()
    with pytest.raises(ValueError):
        update_registered_buffers(foo, 'conv1', ['qweight'], {})