Fix command names

axon/client.py

@@ -575,7 +575,7 @@ def impl_get_task_ip(cluster_name, task_arn, region):
     return nics[0]["Association"]["PublicIp"]
 
 
-def impl_upload_untrained_model_file(model_path, bucket_name, region):
+def impl_upload_untrained_model(model_path, bucket_name, region):
     """
     Uploads an untrained model to S3.
 
@@ -589,7 +589,7 @@ def impl_upload_untrained_model_file(model_path, bucket_name, region):
     print("Uploaded to: {}\n".format(key))
 
 
-def impl_download_untrained_model_file(model_path, bucket_name, region):
+def impl_download_untrained_model(model_path, bucket_name, region):
     """
     Downloads an untrained model from S3.
 
@@ -603,7 +603,7 @@ def impl_download_untrained_model_file(model_path, bucket_name, region):
     print("Downloaded from: {}\n".format(key))
 
 
-def impl_upload_trained_model_file(model_path, bucket_name, region):
+def impl_upload_trained_model(model_path, bucket_name, region):
     """
     Uploads an trained model to S3.
 
@@ -617,7 +617,7 @@ def impl_upload_trained_model_file(model_path, bucket_name, region):
     print("Uploaded to: {}\n".format(key))
 
 
-def impl_download_trained_model_file(model_path, bucket_name, region):
+def impl_download_trained_model(model_path, bucket_name, region):
     """
     Downloads an trained model from S3.
 
@@ -792,56 +792,56 @@ def get_container_ip(task_arn, region):
     print(impl_get_task_ip("axon-autogenerated-cluster", task_arn, region))
 
 
-@cli.command(name="upload-untrained-model-file")
+@cli.command(name="upload-untrained-model")
 @click.argument("model-path")
 @click.option("--region", help="The region to connect to.",
               type=click.Choice(region_choices))
-def upload_untrained_model_file(model_path, region):
+def upload_untrained_model(model_path, region):
     """
     Uploads an untrained model from a local file.
 
     MODEL_PATH The path to the model to upload, ending with the name of the model.
     """
-    impl_upload_untrained_model_file(model_path, ensure_s3_bucket(region), region)
+    impl_upload_untrained_model(model_path, ensure_s3_bucket(region), region)
 
 
-@cli.command(name="download-untrained-model-file")
+@cli.command(name="download-untrained-model")
 @click.argument("model-path")
 @click.option("--region", help="The region to connect to.",
               type=click.Choice(region_choices))
-def download_untrained_model_file(model_path, region):
+def download_untrained_model(model_path, region):
     """
     Downloads an untrained model to a local file.
 
     MODEL_PATH The path to download the model to, ending with the name of the model.
     """
-    impl_download_untrained_model_file(model_path, ensure_s3_bucket(region), region)
+    impl_download_untrained_model(model_path, ensure_s3_bucket(region), region)
 
 
-@cli.command(name="upload-trained-model-file")
+@cli.command(name="upload-trained-model")
 @click.argument("model-path")
 @click.option("--region", help="The region to connect to.",
               type=click.Choice(region_choices))
-def upload_trained_model_file(model_path, region):
+def upload_trained_model(model_path, region):
     """
     Uploads a trained model from a local file.
 
     MODEL_PATH The path to the model to upload, ending with the name of the model.
     """
-    impl_upload_trained_model_file(model_path, ensure_s3_bucket(region), region)
+    impl_upload_trained_model(model_path, ensure_s3_bucket(region), region)
 
 
-@cli.command(name="download-trained-model-file")
+@cli.command(name="download-trained-model")
 @click.argument("model-path")
 @click.option("--region", help="The region to connect to.",
               type=click.Choice(region_choices))
-def download_trained_model_file(model_path, region):
+def download_trained_model(model_path, region):
     """
     Downloads a trained model to a local file.
 
     MODEL_PATH The path to download the model to, ending with the name of the model.
     """
-    impl_download_trained_model_file(model_path, ensure_s3_bucket(region), region)
+    impl_download_trained_model(model_path, ensure_s3_bucket(region), region)
 
 
 @cli.command(name="download-training-script")