fix: update train_config to dev status

PHT-Medic · Mar 7, 2023 · 7e7e2a5 · 7e7e2a5
1 parent bf0ea4b
commit 7e7e2a5
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/train_lib/security/train_config.py b/train_lib/security/train_config.py
@@ -81,6 +81,11 @@ class TrainSource(BaseModel):
     branch: Optional[str] = None
 
 
+class BuildSignature(BaseModel):
+    signature: HexString
+    rsa_public_key: HexString
+
+
 class TrainConfig(BaseModel):
     source: TrainSource
     creator: Creator
@@ -92,6 +97,7 @@ class TrainConfig(BaseModel):
     file_list: List[str]
     hash: HexString  # e_h
     signature: HexString  # e_h_sig
+    build: BuildSignature
     result_hash: Optional[HexString] = None  # e_d
     result_signature: Optional[HexString] = None  # e_d_sig