From 62cc490fd4f4b1c1c5bc07bd8c9a82c6da582e11 Mon Sep 17 00:00:00 2001 From: abuabraham-ttd Date: Tue, 10 Dec 2024 13:44:10 -0800 Subject: [PATCH] Remove aws implemnttion from typedict --- scripts/aws/ec2.py | 8 +++++++- scripts/confidential_compute.py | 4 ---- 2 files changed, 7 insertions(+), 5 deletions(-) diff --git a/scripts/aws/ec2.py b/scripts/aws/ec2.py index 9bb8d42c9..e5f9ebd19 100644 --- a/scripts/aws/ec2.py +++ b/scripts/aws/ec2.py @@ -16,7 +16,13 @@ import yaml sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) -from confidential_compute import ConfidentialCompute, AWSConfidentialComputeConfig, SecretNotFoundException +from confidential_compute import ConfidentialCompute, ConfidentialComputeConfig, SecretNotFoundException + + +class AWSConfidentialComputeConfig(ConfidentialComputeConfig): + enclave_memory_mb: int + enclave_cpu_count: int + class EC2(ConfidentialCompute): diff --git a/scripts/confidential_compute.py b/scripts/confidential_compute.py index 325844513..395c04f49 100644 --- a/scripts/confidential_compute.py +++ b/scripts/confidential_compute.py @@ -12,10 +12,6 @@ class ConfidentialComputeConfig(TypedDict): core_base_url: str optout_base_url: str environment: str - -class AWSConfidentialComputeConfig(ConfidentialComputeConfig): - enclave_memory_mb: int - enclave_cpu_count: int class ConfidentialCompute(ABC):