formatting

jonathf · May 18, 2024 · ab0963d · ab0963d
1 parent 7facef6
commit ab0963d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/chaospy/descriptives/conditional.py b/chaospy/descriptives/conditional.py
@@ -57,7 +57,9 @@ def E_cond(poly, freeze, dist, **kws):
     if not freeze.isconstant():
         poly, freeze = numpoly.align_indeterminants(poly, freeze)
         poly, freeze = numpoly.align_exponents(poly, freeze)
-        freeze = freeze.exponents[numpy.array(freeze.coefficients, dtype=bool)].flatten()
+        freeze = freeze.exponents[
+            numpy.array(freeze.coefficients, dtype=bool)
+        ].flatten()
 
     else:
         freeze = freeze.tonumpy()

diff --git a/chaospy/distributions/sampler/generator.py b/chaospy/distributions/sampler/generator.py
@@ -90,7 +90,7 @@ def generate_samples(order, domain=1, rule="random", antithetic=None):
         order_saved = order
         order = int(numpy.log(order - dim))
         order = order if order > 1 else 1
-        while (order - 1) * 2**(dim - 1) < order_saved:
+        while (order - 1) * 2 ** (dim - 1) < order_saved:
             order += 1
         trans_ = trans
         trans = lambda x_data: trans_(