From 769eba156260ec2819a6b34e4fe02cbbdfc04e62 Mon Sep 17 00:00:00 2001 From: Christopher Harris Date: Mon, 9 Sep 2024 17:05:10 +0000 Subject: [PATCH] finish removing nvtabular related code --- morpheus/utils/downloader.py | 1 - tests/examples/digital_fingerprinting/test_dfp_file_to_df.py | 1 - tests/test_column_info.py | 1 - 3 files changed, 3 deletions(-) diff --git a/morpheus/utils/downloader.py b/morpheus/utils/downloader.py index d352f7ef22..846d73b65a 100644 --- a/morpheus/utils/downloader.py +++ b/morpheus/utils/downloader.py @@ -66,7 +66,6 @@ def __init__(self, download_method: typing.Union[DownloadMethods, str] = DownloadMethods.DASK_THREAD, dask_heartbeat_interval: str = "30s"): - self._merlin_distributed = None self._dask_heartbeat_interval = dask_heartbeat_interval download_method = os.environ.get("MORPHEUS_FILE_DOWNLOAD_TYPE", download_method) diff --git a/tests/examples/digital_fingerprinting/test_dfp_file_to_df.py b/tests/examples/digital_fingerprinting/test_dfp_file_to_df.py index bb5da24aa1..db70e475fe 100644 --- a/tests/examples/digital_fingerprinting/test_dfp_file_to_df.py +++ b/tests/examples/digital_fingerprinting/test_dfp_file_to_df.py @@ -110,7 +110,6 @@ def test_constructor(config: Config): @mock.patch('morpheus.utils.downloader.Distributed') @mock.patch('morpheus.controllers.file_to_df_controller.process_dataframe') def test_get_or_create_dataframe_from_batch_cache_miss(mock_proc_df: mock.MagicMock, - mock_distributed: mock.MagicMock, mock_obf_to_df: mock.MagicMock, mock_dask_cluster: mock.MagicMock, mock_dask_client: mock.MagicMock, diff --git a/tests/test_column_info.py b/tests/test_column_info.py index 4ea8804b78..c40e7854ac 100644 --- a/tests/test_column_info.py +++ b/tests/test_column_info.py @@ -35,7 +35,6 @@ from morpheus.utils.column_info import RenameColumn from morpheus.utils.column_info import StringCatColumn from morpheus.utils.column_info import StringJoinColumn -from morpheus.utils.nvt.schema_converters import create_and_attach_nvt_workflow from morpheus.utils.schema_transforms import process_dataframe