Skip to content

Commit 983c154

Browse files
committed
fix missing memory opts for process actions
1 parent 8a82ee4 commit 983c154

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

python/seldon/cli/cmd_client.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -209,9 +209,14 @@ def get_valid_input_date_string():
209209
job_info["cmd_args"].append("--single-client")
210210
job_info["cmd_args"].append("%CLIENT_NAME%")
211211

212+
conf_data = command_data["conf_data"]
213+
spark_executor_memory = conf_data["spark_executor_memory"]
214+
spark_driver_memory = conf_data["spark_driver_memory"]
212215
input_date_string = get_valid_input_date_string()
213216
replacements = [
214217
("%INPUT_DATE_STRING%", input_date_string),
218+
("%SPARK_EXECUTOR_MEMORY%", spark_executor_memory),
219+
("%SPARK_DRIVER_MEMORY%", spark_driver_memory)
215220
]
216221

217222
def appy_replacements(item):

0 commit comments

Comments
 (0)