4b4c14f308ccbce6d3f1190828a2c8ddaa4752a2,tests/models_tests/prediction_tests/test_set_up_predictor.py,,models_dict,#,29

Before Change

@pytest.fixture
def models_dict():
    // type: () -> Dict[str, chainer.Link]
    return {
        "nfp": NFP,
        "ggnn": GGNN,
        "schnet": SchNet,
        "weavenet": WeaveNet,
        "rsgcn": RSGCN,
        "relgcn": RelGCN,
        "relgat": RelGAT,
        "gin": GIN,
        "nfp_gwm": NFP_GWM,
        "ggnn_gwm": GGNN_GWM,
        "rsgcn_gwm": RSGCN_GWM,
        "gin_gwm": GIN_GWM,
        "gnnfilm": GNNFiLM
    }


def test_setup_predictor(models_dict):
    // type: (Dict[str, chainer.Link]) -> None

After Change

@pytest.fixture
def models_dict():
    // type: () -> Dict[str, chainer.Link]
    return {
        "nfp": NFP,
        "ggnn": GGNN,
        "schnet": SchNet,
        "weavenet": WeaveNet,
        "rsgcn": RSGCN,
        "relgcn": RelGCN,
        "relgat": RelGAT,
        "gin": GIN,
        "nfp_gwm": NFP_GWM,
        "ggnn_gwm": GGNN_GWM,
        "rsgcn_gwm": RSGCN_GWM,
        "gin_gwm": GIN_GWM,
        "gnnfilm": GNNFiLM,
        "nfp_wle": NFP_WLE,
        "ggnn_wle": GGNN_WLE,
        "relgat_wle": RelGAT_WLE,
        "relgcn_wle": RelGCN_WLE,
        "rsgcn_wle": RSGCN_WLE,
        "gin_wle": GIN_WLE,
        "nfp_cwle": NFP_CWLE,
        "ggnn_cwle": GGNN_CWLE,
        "relgat_cwle": RelGAT_CWLE,
        "relgcn_cwle": RelGCN_CWLE,
        "rsgcn_cwle": RSGCN_CWLE,
        "gin_cwle": GIN_CWLE,
        "nfp_gwle": NFP_GWLE,
        "ggnn_gwle": GGNN_GWLE,
        "relgat_gwle": RelGAT_GWLE,
        "relgcn_gwle": RelGCN_GWLE,
        "rsgcn_gwle": RSGCN_GWLE,
        "gin_gwle": GIN_GWLE
    }


def test_setup_predictor(models_dict):
    // type: (Dict[str, chainer.Link]) -> None

Instances