Show gpu form input always

dtaniwaki · Feb 27, 2018 · 732daf1 · 732daf1
1 parent 6ce67fe
commit 732daf1
Showing 1 changed file with 13 additions and 18 deletions.
diff --git 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):
                 </div>
             </div>
         </div>
+        <div class="form-group">
+            <div class="row">
+                <div class="col-sm-4">
+                    <label for="gpu">GPU</label>
+                    <input id="gpu" class="form-control" name="gpu" type="number" step="1"
+                     value="%(gpu)s" min="%(min_gpu)s" max="%(max_gpu)s" required />
+                </div>
+            </div>
+        </div>
         """ % {
             '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 += """
-            <div class="form-group">
-                <div class="row">
-                    <div class="col-sm-4">
-                        <label for="gpu">GPU</label>
-                        <input id="gpu" class="form-control" name="gpu" type="number" step="1"
-                         value="%(gpu)s" min="%(min_gpu)s" max="%(max_gpu)s" required />
-                    </div>
-                </div>
-            </div>
-            """ % {
-                'min_gpu': 0,
-                'max_gpu': self.max_gpu,
-                'gpu': self.stored_user_options.get('gpu', self.gpu),
-            }
         return """<div>%s</div>""" % template
 
     @gen.coroutine