1 files changed, 7 insertions, 6 deletions
diff --git a/app/client/modules/pix2pixhd/views/pix2pixhd.train.js b/app/client/modules/pix2pixhd/views/pix2pixhd.train.js
index 1641976..c5e8e67 100644
--- a/app/client/modules/pix2pixhd/views/pix2pixhd.train.js
+++ b/app/client/modules/pix2pixhd/views/pix2pixhd.train.js
@@ -83,7 +83,7 @@ class Pix2PixHDTrain extends Component {
       if (folder.name === 'results') return
       const datasets = folder.datasets.map(name => {
         const dataset = datasetLookup[name]
-        if (dataset.checkpoints.length) {
+        if (dataset.isBuilt && dataset.checkpoints.length) {
           return name
         }
         return null
@@ -104,10 +104,10 @@ class Pix2PixHDTrain extends Component {
         </div>
         <div className='columns'>
           <div className='column'>
-            <Group title='Dataset'>
+            <Group title='Sequence'>
               <SelectGroup
                 name='checkpoint_name'
-                title='Dataset'
+                title='Sequence name'
                 options={checkpointGroups}
                 onChange={this.handleChange}
                 placeholder='Pick a dataset'
@@ -116,7 +116,7 @@ class Pix2PixHDTrain extends Component {
               <Select
                 title="Epoch"
                 name="epoch"
-                options={this.props.pix2pixhd.data.epochs}
+                options={this.props.pix2pixhd.epochs}
                 onChange={this.handleChange}
                 value={this.state.epoch}
               />
@@ -132,11 +132,12 @@ class Pix2PixHDTrain extends Component {
 
   renderEditor(){
     const { pix2pixhd, queue } = this.props
-    const { checkpoint } = pix2pixhd
+    const { checkpoint, folder_id } = pix2pixhd
     return (
       <div>
         <Group title='Sequence Editor'>
           <SequenceEditor
+            folder_id={folder_id}
             module={pix2pixhdModule}
             checkpoint={checkpoint}
           />
@@ -156,7 +157,7 @@ class Pix2PixHDTrain extends Component {
                   })
                 }}
                 onTrain={() => {
-                  this.props.remote.train_task(this.state.checkpoint_name, pix2pixhd.folder_id, 1)
+                  this.props.remote.train_task(this.state.checkpoint_name, folder_id, 1)
                   setTimeout(() => { // auto-generate epoch demo
                     this.props.remote.augment_task(this.state.checkpoint_name, {
                       ...this.state,