app bugz

1 files changed, 5 insertions, 5 deletions

diff --git a/app/client/dashboard/dashboard.component.js b/app/client/dashboard/dashboard.component.js
index 1e037bc..58795ee 100644
--- a/app/client/dashboard/dashboard.component.js
+++ b/app/client/dashboard/dashboard.component.js
@@ -30,7 +30,7 @@ class Dashboard extends Component {
     //   this.props.actions.list_epochs(nextProps.opt.checkpoint_name)
   }
   render(){
-    const { site, foldersByModule, sampleRNNrenders, queue, images } = this.props
+    const { site, foldersByModule, renders, queue, images } = this.props
     const { tasks } = queue
     const folders = foldersByModule && Object.keys(modules).sort().map(key => {
       if (! foldersByModule[key]) return null
@@ -42,10 +42,10 @@ class Dashboard extends Component {
           </div>
         )
       })
-      const files = key === 'samplernn' && (
+      let files = renders[key] && (
         <FileList
           linkFiles
-          files={sampleRNNrenders}
+          files={renders[key]}
           orderBy='date desc'
           fields={'name date epoch size'}
           onClick={(file, e) => {
@@ -55,7 +55,7 @@ class Dashboard extends Component {
             this.handlePick(file)
           }}
         />
-      )
+      );
 
       return (
         <div className='dashboardModule' key={key}>
@@ -100,7 +100,7 @@ class Dashboard extends Component {
 const mapStateToProps = state => ({
   site: state.system.site,
   foldersByModule: state.dashboard.data.foldersByModule,
-  sampleRNNrenders: state.dashboard.data.sampleRNNrenders,
+  renders: state.dashboard.data.renders,
   images: state.dashboard.images,
   files: state.dashboard.files,
   queue: state.queue,