Cleanup

nv-morpheus · Apr 24, 2024 · 3b6408e · 3b6408e
1 parent 4b771e8
commit 3b6408e
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/morpheus/service/vdb/milvus_vector_db_service.py b/morpheus/service/vdb/milvus_vector_db_service.py
@@ -20,8 +20,6 @@
 import typing
 from functools import wraps
 
-import pandas as pd
-
 import cudf
 
 from morpheus.io.utils import truncate_string_cols_by_bytes
@@ -752,7 +750,7 @@ def create_from_dataframe(self,
         ----------
         name : str
             Name of the collection.
-        df : Union[cudf.DataFrame, pd.DataFrame]
+        df : DataFrameType
             The dataframe to create the collection from.
         overwrite : bool, optional
             Whether to overwrite the collection if it already exists. Default is False.