Update polyaxonfile_declarations.yml
@@ -14,7 +14,7 @@ declarations:
build:
image: tensorflow/tensorflow:1.4.1-py3
build_steps:
- - pip3 install --no-cache-dir -U polyaxon-helper
+ - pip3 install --no-cache-dir -U polyaxon-client
run:
cmd: python3 model.py --batch_size={{ batch_size }} \
