From 01d67db85789f687655139c3825976e95dc58d7f Mon Sep 17 00:00:00 2001 From: David Gardner Date: Mon, 10 Jul 2023 11:57:31 -0700 Subject: [PATCH 1/2] Cleanup imports and type hints --- tests/utils/nvt/test_transforms.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/tests/utils/nvt/test_transforms.py b/tests/utils/nvt/test_transforms.py index 235b5e1d5c..303862b3b7 100644 --- a/tests/utils/nvt/test_transforms.py +++ b/tests/utils/nvt/test_transforms.py @@ -16,9 +16,8 @@ import pytest from nvtabular.ops.operator import ColumnSelector -import cudf - from morpheus.utils.nvt.transforms import json_flatten +from morpheus.utils.type_aliases import DataFrameType from utils.dataset_manager import DatasetManager @@ -37,7 +36,7 @@ def df_fixture(dataset: DatasetManager, data: dict): yield dataset.df_class(data) -def test_json_flatten(df: pd.DataFrame): +def test_json_flatten(df: DataFrameType): col_selector = ColumnSelector(["info"]) result = json_flatten(col_selector, df) From 82e7b050e3148f49f3894888946f55b630b4c512 Mon Sep 17 00:00:00 2001 From: David Gardner Date: Mon, 10 Jul 2023 12:17:13 -0700 Subject: [PATCH 2/2] Use compare to allow for subtle rouding differences --- .../digital_fingerprinting/test_dfp_preprocessing_stage.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/examples/digital_fingerprinting/test_dfp_preprocessing_stage.py b/tests/examples/digital_fingerprinting/test_dfp_preprocessing_stage.py index fd134e649b..4d631fd7d9 100644 --- a/tests/examples/digital_fingerprinting/test_dfp_preprocessing_stage.py +++ b/tests/examples/digital_fingerprinting/test_dfp_preprocessing_stage.py @@ -61,4 +61,4 @@ def test_process_features( results = stage.process_features(dfp_multi_message) assert isinstance(results, MultiDFPMessage) - dataset_pandas.assert_df_equal(results.get_meta_dataframe(), expected_df) + dataset_pandas.assert_compare_df(results.get_meta_dataframe(), expected_df)