Clarification that reweighing algorithm only modifies the instance_we…

…ights attribute (Trusted-AI#159) * Clarified output of Reweighing algorithm's transform() method
Illia-Kryvoviaz · Mar 11, 2020 · c5d88c9 · c5d88c9
1 parent 375c210
commit c5d88c9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/aif360/algorithms/preprocessing/reweighing.py b/aif360/algorithms/preprocessing/reweighing.py
@@ -77,7 +77,8 @@ def transform(self, dataset):
         Args:
             dataset (BinaryLabelDataset): Dataset that needs to be transformed.
         Returns:
-            dataset (BinaryLabelDataset): Transformed dataset.
+            dataset (BinaryLabelDataset): Dataset with transformed
+                instance_weights attribute.
         """
 
         dataset_transformed = dataset.copy(deepcopy=True)