1 files changed, 4 insertions, 3 deletions
diff --git a/app/client/modules/pix2pixhd/pix2pixhd.actions.js b/app/client/modules/pix2pixhd/pix2pixhd.actions.js
index 811f913..a17eeab 100644
--- a/app/client/modules/pix2pixhd/pix2pixhd.actions.js
+++ b/app/client/modules/pix2pixhd/pix2pixhd.actions.js
@@ -202,7 +202,7 @@ export const list_epochs = (checkpoint_name) => (dispatch) => {
   })
 }
 
-export const count_sequence_dataset = (checkpoint_name) => (dispatch) => {
+export const count_dataset = (checkpoint_name) => (dispatch) => {
   const module = pix2pixhdModule.name
   util.allProgress([
     actions.socket.count_directory({ module, dir: 'sequences/' + checkpoint_name + '/' }),
@@ -218,8 +218,9 @@ export const count_sequence_dataset = (checkpoint_name) => (dispatch) => {
     const [sequenceCount, datasetCount] = res //, datasets, results, output, datasetUsage, lossReport] = res
     console.log(sequenceCount, datasetCount)
     dispatch({
-      type: types.pix2pixhd.load_results,
-      results: {
+      type: types.pix2pixhd.load_dataset_count,
+      data: {
+        name: checkpoint_name,
         sequenceCount,
         datasetCount,
       }