From 4c1842b77ba8df42abc33424b0a03722559d8727 Mon Sep 17 00:00:00 2001 From: David Gardner Date: Mon, 12 Aug 2024 14:16:21 -0700 Subject: [PATCH] Fix usage of the C++ impl of write_df_to_file_cpp, prevents the file from being written twice --- morpheus/io/serializers.py | 1 + 1 file changed, 1 insertion(+) diff --git a/morpheus/io/serializers.py b/morpheus/io/serializers.py index 8672383f9e..90822ab6f7 100644 --- a/morpheus/io/serializers.py +++ b/morpheus/io/serializers.py @@ -206,6 +206,7 @@ def write_df_to_file(df: DataFrameType, file_name: str, file_type: FileTypes = F if (CppConfig.get_should_use_cpp() and isinstance(df, cudf.DataFrame)): # Use the C++ implementation write_df_to_file_cpp(df=df, filename=file_name, file_type=file_type, **kwargs) + return mode = file_type