Skip to content
Prev Previous commit
Next Next commit
make format
  • Loading branch information
1vn committed Jul 5, 2019
commit a845387aa04818524ea52af96f04e569a53e78f6
2 changes: 1 addition & 1 deletion pkg/workloads/tf_api/api.py
Original file line number Diff line number Diff line change
Expand Up @@ -379,6 +379,7 @@ def predict(deployment_name, api_name):

return jsonify(response)


def valid_model_dir(model_dir):
"""
validates that model_dir has the expected directory tree, e.g:
Expand Down Expand Up @@ -445,7 +446,6 @@ def start(args):
if not os.path.isdir(args.model_dir):
ctx.storage.download_and_unzip_external(api["model"], args.model_dir)


try:
valid_model_dir(args.model_dir)
except Exception as e:
Expand Down