fix tests

nebari-dev · Nov 13, 2024 · 6b32630 · 6b32630
1 parent 54d691f
commit 6b32630
Showing 1 changed file with 2 additions and 5 deletions.
diff --git 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):