Add arguments to PCF 2 games (#1609)

Summary: Pull Request resolved: #1609 Reviewed By: danbunnell, joe1234wu Differential Revision: D39636836 fbshipit-source-id: fd491ce962aeda46ffc038c7d001840a4168645b
facebookresearch · Sep 21, 2022 · a4ba8dc · a4ba8dc
1 parent 6d8cb9e
commit a4ba8dc
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/fbpcs/private_computation/repository/private_computation_game.py b/fbpcs/private_computation/repository/private_computation_game.py
@@ -64,6 +64,10 @@ class GameNamesValue(TypedDict):
             OneDockerArgument(name="run_name", required=False),
             OneDockerArgument(name="run_id", required=False),
             OneDockerArgument(name="pc_feature_flags", required=False),
+            OneDockerArgument(name="use_tls", required=False),
+            OneDockerArgument(name="ca_cert_path", required=False),
+            OneDockerArgument(name="server_cert_path", required=False),
+            OneDockerArgument(name="private_key_path", required=False),
         ],
     },
     GameNames.SHARD_AGGREGATOR.value: {
@@ -94,6 +98,10 @@ class GameNamesValue(TypedDict):
             OneDockerArgument(name="log_cost", required=False),
             OneDockerArgument(name="run_name", required=False),
             OneDockerArgument(name="visibility", required=False),
+            OneDockerArgument(name="use_tls", required=False),
+            OneDockerArgument(name="ca_cert_path", required=False),
+            OneDockerArgument(name="server_cert_path", required=False),
+            OneDockerArgument(name="private_key_path", required=False),
         ],
     },
     GameNames.DECOUPLED_ATTRIBUTION.value: {
@@ -146,6 +154,10 @@ class GameNamesValue(TypedDict):
             OneDockerArgument(name="use_new_output_format", required=False),
             OneDockerArgument(name="run_id", required=False),
             OneDockerArgument(name="pc_feature_flags", required=False),
+            OneDockerArgument(name="use_tls", required=False),
+            OneDockerArgument(name="ca_cert_path", required=False),
+            OneDockerArgument(name="server_cert_path", required=False),
+            OneDockerArgument(name="private_key_path", required=False),
         ],
     },
     GameNames.PCF2_AGGREGATION.value: {
@@ -166,6 +178,10 @@ class GameNamesValue(TypedDict):
             OneDockerArgument(name="use_new_output_format", required=False),
             OneDockerArgument(name="run_id", required=False),
             OneDockerArgument(name="pc_feature_flags", required=False),
+            OneDockerArgument(name="use_tls", required=False),
+            OneDockerArgument(name="ca_cert_path", required=False),
+            OneDockerArgument(name="server_cert_path", required=False),
+            OneDockerArgument(name="private_key_path", required=False),
         ],
     },
 }