diff --git a/src/pytfe/models/__init__.py b/src/pytfe/models/__init__.py index 8524e6b1..71faee66 100644 --- a/src/pytfe/models/__init__.py +++ b/src/pytfe/models/__init__.py @@ -291,6 +291,19 @@ SSHKeyListOptions, SSHKeyUpdateOptions, ) +from .state_version import ( + StateVersion, + StateVersionCreateOptions, + StateVersionCurrentOptions, + StateVersionList, + StateVersionListOptions, + StateVersionReadOptions, +) +from .state_version_output import ( + StateVersionOutput, + StateVersionOutputsList, + StateVersionOutputsListOptions, +) from .team import ( OrganizationAccess, Team, @@ -652,6 +665,17 @@ "VariableSetVariableCreateOptions", "VariableSetVariableListOptions", "VariableSetVariableUpdateOptions", + # State Versions + "StateVersion", + "StateVersionCreateOptions", + "StateVersionCurrentOptions", + "StateVersionList", + "StateVersionListOptions", + "StateVersionReadOptions", + # State Version Outputs + "StateVersionOutput", + "StateVersionOutputsList", + "StateVersionOutputsListOptions", ] # Rebuild models with forward references after all models are loaded