Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Dec 10, 2024
1 parent f6f7cfb commit 97846e2
Showing 1 changed file with 6 additions and 2 deletions.
8 changes: 6 additions & 2 deletions tests/backends/ott/test_backend.py
Original file line number Diff line number Diff line change
Expand Up @@ -351,7 +351,9 @@ def test_pull(
xx, yy = xy
solver = solver_t()
additional_kwargs = {"alpha": 1.0} if xy is None else {}
out = solver(a=jnp.ones(len(x)) / len(x), b=jnp.ones(len(y)) / len(y), x=x, y=y, xy=(xx, yy), **additional_kwargs)
out = solver(
a=jnp.ones(len(x)) / len(x), b=jnp.ones(len(y)) / len(y), x=x, y=y, xy=(xx, yy), **additional_kwargs
)
p = out.pull(b, scale_by_marginals=False)

assert isinstance(out, BaseDiscreteSolverOutput)
Expand Down Expand Up @@ -404,5 +406,7 @@ def test_plot_errors_sink(self, x: Geom_t, y: Geom_t):
out.plot_errors()

def test_plot_errors_gw(self, x: Geom_t, y: Geom_t):
out = GWSolver(store_inner_errors=True)(a=jnp.ones(len(x)) / len(x), b=jnp.ones(len(y)) / len(y), x=x, y=y, alpha=1.0)
out = GWSolver(store_inner_errors=True)(
a=jnp.ones(len(x)) / len(x), b=jnp.ones(len(y)) / len(y), x=x, y=y, alpha=1.0
)
out.plot_errors()

0 comments on commit 97846e2

Please sign in to comment.