Skip to content

Commit 2805eb7

Browse files
committed
[Scripts] Update CMS exporter: Make current impl. of _get_task_type_parameters specific to protocol version 1
1 parent 1740f8d commit 2805eb7

File tree

1 file changed

+18
-16
lines changed

1 file changed

+18
-16
lines changed

scripts/exporters/CMS.py

+18-16
Original file line numberDiff line numberDiff line change
@@ -98,22 +98,24 @@ def copy_file(self, file, relative_dest):
9898

9999

100100
def _get_task_type_parameters(self, task_data, task_type):
101-
task_type_params = task_data.get("type_params", {})
102-
103-
if task_type == "Communication":
104-
num_processes = task_data.get("num_processes")
105-
if num_processes is not None:
106-
task_type_params["task_type_parameters_Communication_num_processes"] = num_processes
107-
108-
if task_type == "Batch":
109-
HAS_GRADER = get_bool_environ("HAS_GRADER")
110-
if HAS_GRADER:
111-
compilation_type = "grader"
112-
else:
113-
compilation_type = "alone"
114-
task_type_params["task_type_parameters_Batch_compilation"] = compilation_type
115-
116-
return json.dumps(task_type_params)
101+
if self.protocol_version == 1:
102+
task_type_params = task_data.get("type_params", {})
103+
104+
if task_type == "Communication":
105+
num_processes = task_data.get("num_processes")
106+
if num_processes is not None:
107+
task_type_params["task_type_parameters_Communication_num_processes"] = num_processes
108+
109+
if task_type == "Batch":
110+
HAS_GRADER = get_bool_environ("HAS_GRADER")
111+
if HAS_GRADER:
112+
compilation_type = "grader"
113+
else:
114+
compilation_type = "alone"
115+
task_type_params["task_type_parameters_Batch_compilation"] = compilation_type
116+
117+
return json.dumps(task_type_params)
118+
return None
117119

118120

119121
def export_problem_global_data(self):

0 commit comments

Comments
 (0)