Exemplos de RunDBInterface.get_function em Python

Exemplo n.º 1

def test_save_get_function(db: RunDBInterface):
    func, name, proj, tag = {'x': 1, 'y': 2}, 'f1', 'p2', 't3u'
    db.store_function(func, name, proj, tag)
    db_func = db.get_function(name, proj, tag)
    updated = db_func['metadata']['updated']
    if isinstance(db, FileRunDB):
        db_func['metadata']['updated'] = updated.replace(tzinfo=timezone.utc)
    assert func == db_func, 'wrong func'

Exemplo n.º 2

Arquivo: test_dbs.py Projeto: yonittanenbaum/mlrun

def test_save_get_function(db: RunDBInterface):
    func, name, proj, tag = {"x": 1, "y": 2}, "f1", "p2", "t3u"
    db.store_function(func, name, proj, tag)
    db_func = db.get_function(name, proj, tag)

    # db methods enriches metadata
    del db_func["metadata"]
    del func["metadata"]
    assert func == db_func, "wrong func"

Exemplo n.º 3

Arquivo: test_dbs.py Projeto: zhuohuwu0603/mlrun

def test_save_get_function(db: RunDBInterface):
    func, name, proj, tag = {'x': 1, 'y': 2}, 'f1', 'p2', 't3u'
    db.store_function(func, name, proj, tag)
    db_func = db.get_function(name, proj, tag)

    # db methods enriches metadata
    del db_func['metadata']
    del func['metadata']
    assert func == db_func, 'wrong func'