1 files changed, 27 insertions, 11 deletions

diff --git a/public/bundle.js b/public/bundle.js
index d8335da..6710809 100644
--- a/public/bundle.js
+++ b/public/bundle.js
@@ -2192,57 +2192,73 @@ var AugmentationGrid = function (_Component) {
     value: function render() {
       var _this2 = this;
 
+      var _props = this.props,
+          make = _props.make,
+          take = _props.take,
+          checkpoint = _props.checkpoint,
+          onTrain = _props.onTrain,
+          onAugment = _props.onAugment;
+      var _state = this.state,
+          x = _state.x,
+          y = _state.y,
+          sum = _state.sum;
+
       var rows = [];
       return (0, _preact.h)(
         _group2.default,
         { className: 'augmentationGrid' },
         (0, _preact.h)(_buttonGrid2.default, {
-          x: this.props.make,
-          y: this.props.take,
+          x: make,
+          y: take,
           max: 5000,
           onHover: function onHover(x, y) {
             return _this2.setState({ x: x, y: y });
           },
           onClick: function onClick(x, y) {
-            _this2.setState({ sum: _this2.state.sum + x * y });
-            _this2.props.onAugment(y, x);
+            _this2.setState({ sum: sum + x * y });
+            onAugment(y, x);
           }
         }),
         (0, _preact.h)(
           _param2.default,
+          { title: 'Name' },
+          checkpoint.name
+        ),
+        (0, _preact.h)(
+          _param2.default,
           { title: 'Take' },
-          this.state.y
+          y
         ),
         (0, _preact.h)(
           _param2.default,
           { title: 'Make' },
-          this.state.x
+          x
         ),
         (0, _preact.h)(
           _param2.default,
           { title: 'Will add to dataset' },
-          this.state.x * this.state.y
+          x * y
         ),
         (0, _preact.h)(
           _param2.default,
           { title: 'Total added this epoch' },
-          this.state.sum
+          sum
         ),
         (0, _preact.h)(
           _param2.default,
           { title: 'Sequence length' },
-          this.state.checkpoint.sequenceCount
+          checkpoint.sequenceCount
         ),
         (0, _preact.h)(
           _param2.default,
           { title: 'Dataset size' },
-          this.state.checkpoint.datasetCount
+          checkpoint.datasetCount
         ),
         (0, _preact.h)(
           _button2.default,
           { onClick: function onClick() {
               _this2.setState({ sum: 0 });
-              _this2.props.onTrain();
+              onTrain();
             } },
           'Train'
         )