Unverified Commit b29b7e55 authored by SparkSnail's avatar SparkSnail Committed by GitHub
Browse files

Add exception handler in trial_keeper (#235)

* add exception handling in trial_keeper.py
parent 5c634149
...@@ -54,8 +54,8 @@ def main_loop(args): ...@@ -54,8 +54,8 @@ def main_loop(args):
print('subprocess terminated. Exit code is {}. Quit'.format(retCode)) print('subprocess terminated. Exit code is {}. Quit'.format(retCode))
#copy local directory to hdfs #copy local directory to hdfs
nni_local_output_dir = os.environ['NNI_OUTPUT_DIR'] nni_local_output_dir = os.environ['NNI_OUTPUT_DIR']
hdfs_client = HdfsClient(hosts='{0}:{1}'.format(args.pai_hdfs_host, '50070'), user_name=args.pai_user_name, timeout=5)
try: try:
hdfs_client = HdfsClient(hosts='{0}:{1}'.format(args.pai_hdfs_host, '50070'), user_name=args.pai_user_name, timeout=5)
if copyDirectoryToHdfs(nni_local_output_dir, args.pai_hdfs_output_dir, hdfs_client): if copyDirectoryToHdfs(nni_local_output_dir, args.pai_hdfs_output_dir, hdfs_client):
print('copy directory from {0} to {1} success!'.format(nni_local_output_dir, args.pai_hdfs_output_dir)) print('copy directory from {0} to {1} success!'.format(nni_local_output_dir, args.pai_hdfs_output_dir))
else: else:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment