diff --git a/src/_nebari/stages/infrastructure/__init__.py b/src/_nebari/stages/infrastructure/__init__.py index 0fccd1bbfb..f430c49126 100644 --- a/src/_nebari/stages/infrastructure/__init__.py +++ b/src/_nebari/stages/infrastructure/__init__.py @@ -57,11 +57,6 @@ class DigitalOceanInputVars(schema.Base): kubeconfig_filename: str = get_kubeconfig_filename() -class GCPGuestAccelerators(schema.Base): - name: str - count: int - - class GCPNodeGroupInputVars(schema.Base): name: str instance_type: str @@ -69,7 +64,7 @@ class GCPNodeGroupInputVars(schema.Base): max_size: int labels: Dict[str, str] preemptible: bool - guest_accelerators: List[GCPGuestAccelerators] + guest_accelerators: List["GCPGuestAccelerator"] class GCPPrivateClusterConfig(schema.Base):