[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

src/stream_mapper/pytorch/_base.py

@@ -13,7 +13,7 @@
 from stream_mapper.core import ModelBase as CoreModelBase
 from stream_mapper.core._connect.nn_namespace import NN_NAMESPACE
 from stream_mapper.core._connect.xp_namespace import XP_NAMESPACE
-from stream_mapper.core.typing import ArrayNamespace  # noqa: TCH001
+from stream_mapper.core.typing import ArrayNamespace  # noqa: TC001
 from stream_mapper.core.utils.dataclasses import ArrayNamespaceReprMixin
 from stream_mapper.core.utils.scale import names_intersect
 

src/stream_mapper/pytorch/builtin/compat/__init__.py

@@ -4,4 +4,4 @@
 from stream_mapper.pytorch.builtin.compat.nflow import NFlowModel
 from stream_mapper.pytorch.builtin.compat.zuko import ZukoFlowModel
 
-__all__ = ("ZukoFlowModel", "NFlowModel", "KDEModel")
+__all__ = ("KDEModel", "NFlowModel", "ZukoFlowModel")

src/stream_mapper/pytorch/builtin/compat/kde.py

@@ -4,7 +4,7 @@
 
 __all__: tuple[str, ...] = ()
 
-from collections.abc import Callable  # noqa: TCH003
+from collections.abc import Callable  # noqa: TC003
 from dataclasses import KW_ONLY, dataclass
 from typing import TYPE_CHECKING
 

src/stream_mapper/pytorch/prior.py

@@ -8,7 +8,7 @@
 from stream_mapper.core.prior import *  # noqa: F403
 from stream_mapper.core.prior._track import ControlRegions as CoreControlRegions
 from stream_mapper.core.prior._weight import HardThreshold as CoreHardThreshold
-from stream_mapper.core.typing import ArrayNamespace  # noqa: TCH001
+from stream_mapper.core.typing import ArrayNamespace  # noqa: TC001
 
 from stream_mapper.pytorch.typing import Array
 

src/stream_mapper/pytorch/typing.py

@@ -1,6 +1,6 @@
 """Pytorch type hints."""
 
-__all__ = ("Array", "NNModel", "ArrayNamespace")
+__all__ = ("Array", "ArrayNamespace", "NNModel")
 
 from torch import Tensor as Array
 from torch.nn import Module as NNModel