added docs to drop.py but why do we have both drop and keep they seem to be doing the same thing from different directions

Author: eodole

bayesflow/adapters/transforms/drop.py

@@ -11,6 +11,26 @@
 
 @serializable(package="bayesflow.adapters")
 class Drop(Transform):
+    """ 
+    Transform to drop variables from further calculation. 
+
+    Parameters: 
+        keys: list of strings, containing names of data variables that should be dropped 
+    
+    Example: 
+
+    >>> import bayesflow as bf
+    >>> a = [1,2,3,4]
+    >>> b = [[1,2],[3,4]]
+    >>> c = [[5,6,7,8]]
+    >>> dat = dict(a=a,b=b,c=c)
+    >>> dat
+        {'a': [1, 2, 3, 4], 'b': [[1, 2], [3, 4]], 'c': [[5, 6, 7, 8]]}
+    >>> drop = bf.adapters.transforms.Drop(("b","c"))
+    >>> drop.forward(dat)
+        {'a': [1, 2, 3, 4]}
+    """
+
     def __init__(self, keys: Sequence[str]):
         self.keys = keys