diff --git a/ucrspawner/ucrspawner.py b/ucrspawner/ucrspawner.py
index 19937b2..df42d52 100644
--- a/ucrspawner/ucrspawner.py
+++ b/ucrspawner/ucrspawner.py
@@ -267,8 +267,7 @@ def options_from_form(self, formdata):
options['cpu'] = float(formdata['cpu'][0])
options['mem'] = float(formdata['mem'][0])
options['disk'] = float(formdata['disk'][0])
- if formdata.get('gpu', None):
- options['gpu'] = int(formdata['gpu'][0])
+ options['gpu'] = int(formdata['gpu'][0])
return options
@property
@@ -304,6 +303,15 @@ def options_form(self):
+
""" % {
'default_app_image': self.app_image,
'app_image': self.stored_user_options.get('app_image', None) or '',
@@ -316,23 +324,10 @@ def options_form(self):
'min_disk': 1000.0,
'max_disk': self.max_disk,
'disk': remove_zeros(str(self.stored_user_options.get('disk', self.disk))),
+ 'min_gpu': 0,
+ 'max_gpu': self.max_gpu,
+ 'gpu': self.stored_user_options.get('gpu', self.gpu),
}
- if self.max_gpu > 0:
- template += """
-
- """ % {
- 'min_gpu': 0,
- 'max_gpu': self.max_gpu,
- 'gpu': self.stored_user_options.get('gpu', self.gpu),
- }
return """%s
""" % template
@gen.coroutine