Implement backend.abc (#716)

probdiffeq/backend/abc.py

@@ -0,0 +1,12 @@
+"""Abstract base classes."""
+
+
+import abc
+
+
+class ABC(abc.ABC):  # noqa: B024,D101
+    pass
+
+
+def abstractmethod(*args):
+    return abc.abstractmethod(*args)

probdiffeq/impl/_conditional.py

@@ -1,5 +1,5 @@
 """Conditionals."""
-import abc
+from probdiffeq.backend import abc
 
 
 class ConditionalBackend(abc.ABC):

probdiffeq/impl/_hidden_model.py

@@ -1,4 +1,4 @@
-import abc
+from probdiffeq.backend import abc
 
 
 class HiddenModelBackend(abc.ABC):

probdiffeq/impl/_impl.py

@@ -1,8 +1,8 @@
 """State-space model impl."""
-import abc
 import warnings
 from typing import Optional
 
+from probdiffeq.backend import abc
 from probdiffeq.impl import (
     _conditional,
     _hidden_model,

probdiffeq/impl/_linearise.py

@@ -1,4 +1,4 @@
-import abc
+from probdiffeq.backend import abc
 
 
 class LinearisationBackend(abc.ABC):

probdiffeq/impl/_prototypes.py

@@ -1,4 +1,4 @@
-import abc
+from probdiffeq.backend import abc
 
 
 class PrototypeBackend(abc.ABC):

probdiffeq/impl/_ssm_util.py

@@ -1,6 +1,6 @@
 """SSM utilities."""
 
-import abc
+from probdiffeq.backend import abc
 
 
 class SSMUtilBackend(abc.ABC):

probdiffeq/impl/_stats.py

@@ -1,4 +1,4 @@
-import abc
+from probdiffeq.backend import abc
 
 
 class StatsBackend(abc.ABC):

probdiffeq/impl/_transform.py

@@ -1,4 +1,4 @@
-import abc
+from probdiffeq.backend import abc
 
 
 class TransformBackend(abc.ABC):

probdiffeq/impl/_variable.py

@@ -1,4 +1,4 @@
-import abc
+from probdiffeq.backend import abc
 
 
 class VariableBackend(abc.ABC):

probdiffeq/solvers/_solver.py

@@ -1,10 +1,10 @@
 """IVP-solver API."""
 
 
-import abc
 from typing import Generic, TypeVar
 
 from probdiffeq import _interp
+from probdiffeq.backend import abc
 from probdiffeq.backend import numpy as np
 from probdiffeq.impl import impl
 

probdiffeq/solvers/calibrated.py

@@ -1,8 +1,6 @@
 """Calibrated IVP solvers."""
-import abc
-
 from probdiffeq import _interp
-from probdiffeq.backend import tree_util
+from probdiffeq.backend import abc, tree_util
 from probdiffeq.impl import impl
 from probdiffeq.solvers import _common, _solver
 

probdiffeq/solvers/strategies/components/corrections.py

@@ -1,8 +1,6 @@
 """Correction models."""
 
-import abc
-
-from probdiffeq.backend import functools
+from probdiffeq.backend import abc, functools
 from probdiffeq.backend import numpy as np
 from probdiffeq.impl import impl
 from probdiffeq.solvers.strategies.components import cubature

probdiffeq/solvers/strategies/strategy.py

@@ -1,10 +1,9 @@
 """Interface for estimation strategies."""
 
-import abc
 from typing import Any, Generic, TypeVar
 
 from probdiffeq import _interp
-from probdiffeq.backend import containers, tree_util
+from probdiffeq.backend import abc, containers, tree_util
 from probdiffeq.backend import numpy as np
 from probdiffeq.impl import impl