Docstring for base argument

alexhernandezgarcia · Jul 10, 2024 · 910a948 · 910a948
1 parent 9520315
commit 910a948
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/gflownet/policy/base.py b/gflownet/policy/base.py
@@ -35,6 +35,8 @@ def __init__(
             The device to be passed to torch tensors.
         float_precision : int or torch.dtype
             The floating point precision to be passed to torch tensors.
+        base: Policy (optional)
+            A base policy to be used as backbone for the backward policy.
         """
         config = self._get_config(config)
         # Device and float precision