diff --git a/normalizing_flows/bijections/__init__.py b/normalizing_flows/bijections/__init__.py index 74bbf98..6bcd91f 100644 --- a/normalizing_flows/bijections/__init__.py +++ b/normalizing_flows/bijections/__init__.py @@ -13,3 +13,4 @@ from normalizing_flows.bijections.finite.residual.architectures import InvertibleResNet, ResFlow, ProximalResFlow from normalizing_flows.bijections.finite.residual.proximal import ProximalResFlowBlock from normalizing_flows.bijections.finite.linear import LowerTriangular, Orthogonal, LU, QR +from normalizing_flows.bijections.finite.linear import Identity \ No newline at end of file