Use the original argument name image_name instead of image for train | predict | evaluate. (#2092)

* Add the version for the docker python package.

* Add the method declaration for the argument parser.

* Add the arguments for train|predict|evaluate

* Copy common modules from elasticdl main package to elasticdl_client package.

* Call _submit_job in train api

* Add k8s_resource module.

* Set default value for extra_pypi_index arguments.

* Update the default base image value to be python:3.7

* Update the argument description.

* Make --image arguments to be required

* Add the implementation of evaluate and predict

* Update the default base image to be python:3.6

* Remove clean command.

* Add comments.

* Rename model_utils to module_utils

* Simplify the function in k8s_client in elasticdl_client package.

* Remove unnecessary event_cb in k8s_client in elasticdl_client package.

* Use argument name image_name instead of image for train|predict|evaluate.

Author: brightcoder01

elasticdl_client/api.py

@@ -106,7 +106,7 @@ def push_zoo(args):
 def train(args):
     container_args = [
         "--worker_image",
-        args.image,
+        args.image_name,
         "--model_zoo",
         args.model_zoo,
         "--cluster_spec",
@@ -126,13 +126,13 @@ def train(args):
         )
     )
 
-    _submit_job(args.image, args, container_args)
+    _submit_job(args.image_name, args, container_args)
 
 
 def evaluate(args):
     container_args = [
         "--worker_image",
-        args.image,
+        args.image_name,
         "--model_zoo",
         args.model_zoo,
         "--cluster_spec",
@@ -151,13 +151,13 @@ def evaluate(args):
         )
     )
 
-    _submit_job(args.image, args, container_args)
+    _submit_job(args.image_name, args, container_args)
 
 
 def predict(args):
     container_args = [
         "--worker_image",
-        args.image,
+        args.image_name,
         "--model_zoo",
         args.model_zoo,
         "--cluster_spec",
@@ -176,7 +176,7 @@ def predict(args):
         )
     )
 
-    _submit_job(args.image, args, container_args)
+    _submit_job(args.image_name, args, container_args)
 
 
 def _submit_job(image_name, client_args, container_args):

elasticdl_client/common/args.py

@@ -208,7 +208,7 @@ def add_common_params(parser):
     """Common arguments for training/prediction/evaluation"""
     add_common_args_between_master_and_worker(parser)
     parser.add_argument(
-        "--image",
+        "--image_name",
         type=str,
         required=True,
         help="The docker image for this job.",

elasticdl_client/common/k8s_client.py

@@ -99,6 +99,9 @@ def __init__(
             cluster_spec_module = load_module(cluster_spec)
             self.cluster = cluster_spec_module.cluster
 
+    def get_master_pod_name(self):
+        return get_master_pod_name(self.job_name)
+
     def patch_labels_to_pod(self, pod_name, labels_dict):
         body = {"metadata": {"labels": labels_dict}}
         try: