add feature transformation

awslabs · Nov 5, 2024 · 7079f2c · 7079f2c
1 parent 6eef513
commit 7079f2c
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/...processing/data_transformations/dist_transformations/dist_hard_negative_transformation.py b/...processing/data_transformations/dist_transformations/dist_hard_negative_transformation.py
@@ -36,13 +36,13 @@ def apply_transform(
     Parameters
     ----------
     cols : Sequence[str]
-        List of column names to apply normalization to.
+        List of column names to apply normalization to
     separator: str, optional
-        The separator for string input value. Only required when input value type is string.
+        The separator for string input value. Only required when input value type is string
     spark: SparkSession
         The spark session
     input_df : DataFrame
-        The input DataFrame to apply transformation to.
+        The input DataFrame to apply transformation to
     edge_mapping_dict: dict
         The mapping dictionary contain mapping file directory and edge type
     """
@@ -86,7 +86,7 @@ class DistHardNegativeTransformation(DistributedTransformation):
     Parameters
     ----------
     separator: str, optional
-        The separator for string input value. Only required when input value type is string.
+        The separator for string input value. Only required when input value type is string
     spark: SparkSession
         The spark session
     edge_mapping_dict: dict