Unverified Commit 69773a6b authored by Suzukazole's avatar Suzukazole
Browse files

more docs and lint

parent 7da392a6
Loading
Loading
Loading
Loading
+10 −5
Original line number Diff line number Diff line
@@ -2478,16 +2478,21 @@ class RxnSplitTransformer(Transformer):
  The input is expected to be in the form reactant>reagent>product. The source
  string would be reactants>reagents and the target string would be the products.

  The transformer can also separate the reagents from the reactants for a mixed
  training mode. This can be toggled (default True) by setting the value of
  sep_reagent while calling the transformer.

  Examples
  -----
  --------

  Note
  -----
  ----
  This class only transforms the feature field of a reaction dataset like USPTO.

  """

  def __init__(self, sep_reagent: bool = True, dataset: Optional[Dataset] = None):
  def __init__(self,
               sep_reagent: bool = True,
               dataset: Optional[Dataset] = None):
    """Initializes the Reaction split Transformer.

    Parameters