Merge pull request #43 from littleape1022/master

Debug: update base_pack_op
ucloud · Sep 17, 2018 · ed0f534 · ed0f534
2 parents 2f0e1c5 + 27cfee2
commit ed0f534
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/uaitrain/operation/pack_docker_image/base_pack_op.py b/uaitrain/operation/pack_docker_image/base_pack_op.py
@@ -37,7 +37,7 @@ def _add_pack_args(self, pack_parser):
         pack_parser.add_argument(
             '--os',
             type=str,
-            default='ubuntu',
+            default='ubuntu-14.04.05',
             help='The docker os version')
         pack_parser.add_argument(
             '--python_version',
@@ -47,6 +47,7 @@ def _add_pack_args(self, pack_parser):
         pack_parser.add_argument(
             '--ai_arch_v',
             type=str,
+            required=True,
             help='The AI framework and its version, e.g., tensorflow-1.1.0')
         pack_parser.add_argument(
             '--acc_type',
@@ -195,7 +196,7 @@ def _translate_pkg_to_id(self, pkgtype, pkg):
         if succ is False:
             raise RuntimeError("Error get {0} info from server".format(pkgtype))
 
-        for avpkg in result['PkgSet']:
+        for avpkg in result['DataSet']:
             if pkgtype == 'OS' or pkgtype == 'Python' or pkgtype == 'AIFrame':
                 versionsplit = pkg.rfind('-')
                 if versionsplit > 0: