From e4d179fe3a0b5dc218b096fb91a9501281716ab8 Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Tue, 7 Jan 2025 23:14:07 +0000
Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 src/moscot/neural/base/problems/__init__.py |  2 +-
 src/moscot/neural/base/problems/problem.py  | 18 ++++--------------
 2 files changed, 5 insertions(+), 15 deletions(-)

diff --git a/src/moscot/neural/base/problems/__init__.py b/src/moscot/neural/base/problems/__init__.py
index 9f9185ea..ec15beb2 100644
--- a/src/moscot/neural/base/problems/__init__.py
+++ b/src/moscot/neural/base/problems/__init__.py
@@ -1,3 +1,3 @@
 from moscot.neural.base.problems.problem import NeuralOTProblem
 
-__all__ = ["NeuralOTProblem"]
\ No newline at end of file
+__all__ = ["NeuralOTProblem"]
diff --git a/src/moscot/neural/base/problems/problem.py b/src/moscot/neural/base/problems/problem.py
index 0778ac1e..cc142f98 100644
--- a/src/moscot/neural/base/problems/problem.py
+++ b/src/moscot/neural/base/problems/problem.py
@@ -12,7 +12,6 @@
     Union,
 )
 
-
 import numpy as np
 import pandas as pd
 
@@ -20,13 +19,9 @@
 
 from moscot import backends
 from moscot._types import ArrayLike, Device_t
-from moscot.base.output import (
-    BaseNeuralOutput,
-)
-from moscot.base.problems._utils import (
-    wrap_prepare,
-    wrap_solve,
-)
+from moscot.base.output import BaseNeuralOutput
+from moscot.base.problems._utils import wrap_prepare, wrap_solve
+from moscot.base.problems.problem import BaseProblem
 from moscot.base.solver import OTSolver
 from moscot.utils.subset_policy import (  # type:ignore[attr-defined]
     ExplicitPolicy,
@@ -35,12 +30,7 @@
     SubsetPolicy,
     create_policy,
 )
-from moscot.utils.tagged_array import (
-    DistributionCollection,
-    DistributionContainer,
-)
-
-from moscot.base.problems.problem import BaseProblem
+from moscot.utils.tagged_array import DistributionCollection, DistributionContainer
 
 K = TypeVar("K", bound=Hashable)