[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

@@ -1,3 +1,3 @@
 from moscot.neural.base.problems.problem import NeuralOTProblem
 
-__all__ = ["NeuralOTProblem"]
+__all__ = ["NeuralOTProblem"]

src/moscot/neural/base/problems/problem.py

@@ -12,21 +12,16 @@
     Union,
 )
 
-
 import numpy as np
 import pandas as pd
 
 from anndata import AnnData
 
 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)