diff --git a/torchflows/bijections/finite/multiscale/architectures.py b/torchflows/bijections/finite/multiscale/architectures.py index f9d6d88..ed32624 100644 --- a/torchflows/bijections/finite/multiscale/architectures.py +++ b/torchflows/bijections/finite/multiscale/architectures.py @@ -4,7 +4,7 @@ from torchflows.bijections.finite.autoregressive.transformers.linear.affine import Shift, Affine from torchflows.bijections.finite.autoregressive.transformers.spline.rational_quadratic import RationalQuadratic -from torchflows.bijections.finite.autoregressive.transformers.spline.linear import Linear as LinearRational +from torchflows.bijections.finite.autoregressive.transformers.spline.linear_rational import LinearRational from torchflows.bijections.finite.autoregressive.transformers.combination.sigmoid import ( DeepSigmoid, DeepDenseSigmoid,