@@ -22,7 +22,7 @@ def print_header(s):
22
22
click .secho (s , fg = 'yellow' , bold = True )
23
23
24
24
25
- def build_kernel (name , tag , extra_opts = '' , * , squash = False ):
25
+ def build_kernel (name , tag , extra_opts = '' , * , squash = False , testing = False ):
26
26
assert Path (name ).is_dir ()
27
27
28
28
sq = '--squash' if squash else ''
@@ -44,11 +44,15 @@ def build_kernel(name, tag, extra_opts='', *, squash=False):
44
44
'the new kernelspec.' ,
45
45
fg = 'yellow' , err = True )
46
46
return
47
+ if testing :
48
+ img_ref = f'beta.docker.backend.ai/testing/{ short_name } :{ tag } '
49
+ else :
50
+ img_ref = f'lablup/{ short_name } :{ tag } '
47
51
run ('docker build '
48
- f'-t lablup/ { short_name } : { tag } { extra_opts } '
52
+ f'-t { img_ref } { extra_opts } '
49
53
f'-f { dockerfile } { sq } { name } ' )
50
54
if auto_push .get ():
51
- run (f'docker push lablup/ { short_name } : { tag } ' )
55
+ run (f'docker push { img_ref } ' )
52
56
53
57
54
58
def build_common (name , tag , extra_opts = '' ):
@@ -74,7 +78,7 @@ def build_common(name, tag, extra_opts=''):
74
78
'tf-old' , 'tf-current' , 'tf-future' ,
75
79
'caffe' , 'pytorch' ,
76
80
'cntk-builder' ,'cntk' ,
77
- 'vendor-aws' , 'vendor-ngc' ,
81
+ 'vendor-aws' , 'vendor-ngc' , 'vendor-freecad' ,
78
82
]
79
83
80
84
@@ -338,6 +342,9 @@ def main(build, list_builds, _auto_push):
338
342
if 'vendor-aws' in build :
339
343
build_kernel ('vendor/aws_polly' , '0.1-alpine3.8' )
340
344
345
+ if 'vendor-freecad' in build :
346
+ build_kernel ('vendor/freecad' , '0.18-glvnd1.1' )
347
+
341
348
if 'vendor-ngc' in build :
342
349
build_kernel ('vendor/ngc-caffe2' , '18.08-py2' )
343
350
build_kernel ('vendor/ngc-caffe3' , '18.08-py3' )
0 commit comments