diff --git a/examples/gnn_fraud_detection_pipeline/stages/model.py b/examples/gnn_fraud_detection_pipeline/stages/model.py index eac1a6f2e6..d82419e87e 100644 --- a/examples/gnn_fraud_detection_pipeline/stages/model.py +++ b/examples/gnn_fraud_detection_pipeline/stages/model.py @@ -165,7 +165,7 @@ def inference(self, """ # create sampler and test dataloaders - full_sampler = dgl.dataloading.MultiLayerNeighborSampler(fanouts=[4, 3]) + full_sampler = dgl.dataloading.MultiLayerFullNeighborSampler(num_layers=3) test_dataloader = dgl.dataloading.DataLoader(input_graph, {target_node: test_idx}, full_sampler, batch_size=batch_size, diff --git a/tests/examples/gnn_fraud_detection_pipeline/test_graph_sage_stage.py b/tests/examples/gnn_fraud_detection_pipeline/test_graph_sage_stage.py index a4273ca17b..886c339962 100644 --- a/tests/examples/gnn_fraud_detection_pipeline/test_graph_sage_stage.py +++ b/tests/examples/gnn_fraud_detection_pipeline/test_graph_sage_stage.py @@ -22,9 +22,9 @@ from morpheus.messages import MessageMeta from morpheus.messages import MultiMessage -# pylint: disable=no-name-in-module - +# pylint: disable=no-name-in-module +@pytest.mark.usefixtures("manual_seed") @pytest.mark.use_python class TestGraphSageStage: @@ -68,4 +68,4 @@ def test_process_message(self, cols = results.inductive_embedding_column_names + ['index'] assert sorted(cols) == sorted(expected_df.columns) ind_emb_df = results.get_meta(cols) - dataset_pandas.assert_compare_df(ind_emb_df.to_pandas(), expected_df) + dataset_pandas.assert_compare_df(ind_emb_df.to_pandas(), expected_df, abs_tol=1, rel_tol=1) diff --git a/tests/tests_data/examples/gnn_fraud_detection_pipeline/inductive_emb.csv b/tests/tests_data/examples/gnn_fraud_detection_pipeline/inductive_emb.csv index 9ae8600dd2..8c1309ce6b 100644 --- a/tests/tests_data/examples/gnn_fraud_detection_pipeline/inductive_emb.csv +++ b/tests/tests_data/examples/gnn_fraud_detection_pipeline/inductive_emb.csv @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:e5f824196fdc9a0e027d65369ae7f73e44fe2986058cc1af46c3c04d62217f74 -size 8641 +oid sha256:aa6fb37a801c5baf58d9512050671e55e21d132c7e56784d49f9910c7d0a3a26 +size 7525