From 6b3263096039d23f043fa946124b323cefab194e Mon Sep 17 00:00:00 2001 From: viniciusdc Date: Tue, 12 Nov 2024 16:47:39 -0300 Subject: [PATCH] fix tests --- src/_nebari/stages/infrastructure/__init__.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/src/_nebari/stages/infrastructure/__init__.py b/src/_nebari/stages/infrastructure/__init__.py index 016f8d3d65..da7c10157b 100644 --- a/src/_nebari/stages/infrastructure/__init__.py +++ b/src/_nebari/stages/infrastructure/__init__.py @@ -8,7 +8,7 @@ import tempfile from typing import Annotated, Any, Dict, List, Literal, Optional, Tuple, Type, Union -from pydantic import Field, field_validator, model_validator +from pydantic import Field, PrivateAttr, field_validator, model_validator from _nebari import constants from _nebari.provider import terraform @@ -135,11 +135,8 @@ class AWSAmiTypes(str, enum.Enum): class AWSNodeLaunchTemplate(schema.Base): - class Config: - underscore_attrs_are_private = True - pre_bootstrap_command: Optional[str] = None - _ami_id: Optional[str] = None + _ami_id: Optional[str] = PrivateAttr(default=None) class AWSNodeGroupInputVars(schema.Base):