black, isort

alexhernandezgarcia · Feb 9, 2024 · c8c1613 · c8c1613
1 parent 67cc2d7
commit c8c1613
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/gflownet/gflownet.py b/gflownet/gflownet.py
@@ -23,13 +23,13 @@
 from gflownet.utils.buffer import Buffer
 from gflownet.utils.common import (
     batch_with_rest,
+    bootstrap_samples,
     set_device,
     set_float_precision,
     tbool,
     tfloat,
     tlong,
     torch2np,
-    bootstrap_samples
 )
 
 
@@ -877,7 +877,10 @@ def estimate_logprobs_data(
             each data point.
 
         logprobs_std: torch.tensor
-            Bootstrap variances of the logprobs_estimates
+            Bootstrap std of the logprobs_estimates
+
+        probs_std: torch.tensor
+            Bootstrap std of the torch.exp(logprobs_estimates)
         """
         print("Compute logprobs...", flush=True)
         times = {}

diff --git a/gflownet/utils/common.py b/gflownet/utils/common.py
@@ -291,6 +291,7 @@ def chdir_random_subdir():
     os.mkdir(cwd)
     os.chdir(cwd)
 
+
 def bootstrap_samples(tensor, num_samples):
     """
     Bootstraps tensor along the last dimention