Remove aws implemnttion from typedict

IABTechLab · Dec 10, 2024 · 62cc490 · 62cc490
1 parent 4c694e7
commit 62cc490
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git 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
@@ -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):