biggan path

2 files changed, 4 insertions, 4 deletions

diff --git a/app/client/modules/biggan/biggan.tasks.js b/app/client/modules/biggan/biggan.tasks.js
index 2aed088..a28bf20 100644
--- a/app/client/modules/biggan/biggan.tasks.js
+++ b/app/client/modules/biggan/biggan.tasks.js
@@ -7,13 +7,11 @@ import actions from '../../actions'
 
 import module from './biggan.module'
 
-export const invert_task = (dataset, folder_id, epochs=1) => dispatch => {
-  dataset = dataset.name || dataset
+export const invert_task = (folder_id) => dispatch => {
   if (dataset === 'PLACEHOLDER') return
   const task = {
     module: module.name,
     activity: 'invert',
-    dataset,
     opt: {
       folder_id: folder_id,
     }
diff --git a/app/client/modules/biggan/views/biggan.show.js b/app/client/modules/biggan/views/biggan.show.js
index 8d85f87..22cecfb 100644
--- a/app/client/modules/biggan/views/biggan.show.js
+++ b/app/client/modules/biggan/views/biggan.show.js
@@ -35,7 +35,7 @@ class BigGANShow extends Component {
     }
   }
   render(){
-    const { biggan, match, history } = this.props
+    const { biggan, match, history, actions } = this.props
     const { folderLookup } = (biggan.data || {})
     const folder = (folderLookup || {})[biggan.folder_id] || {}
 
@@ -45,6 +45,7 @@ class BigGANShow extends Component {
           <div class='spaced'>
             <h1>{folder ? folder.name : <Loading />}</h1>
             <UploadStatus />
+            <button onClick={() => actions.tasks.invert_task(folder.id)}>Run inversion task</button>
           </div>
         </div>
         <div className='row'>
@@ -113,6 +114,7 @@ const mapStateToProps = state => ({
 
 const mapDispatchToProps = (dispatch, ownProps) => ({
   actions: bindActionCreators(bigganActions, dispatch),
+  tasks: bindActionCreators(bigganTasks, dispatch),
   remote: bindActionCreators(bigganTasks, dispatch),
 })