Test

Author: Keivan Vosoughi

app/client/src/pages/DataGenerator/DataGenerator.tsx

@@ -93,17 +93,6 @@ const DataGenerator: FunctionComponent<Props> = () => {
         if (generate_file_name && !mutation.data) {
             mutation.mutate(generate_file_name);
         }
-
-        if (mutation.data && mutation?.data?.dataset) {
-            const dataset = mutation?.data?.dataset as any;
-            form.setFieldsValue({
-                ...initialData,
-                ...dataset,
-                workflow_type: dataset.technique === 'freeform' ? 
-                    WorkflowType.FREE_FORM_DATA_GENERATION : WorkflowType.CUSTOM_DATA_GENERATION   
-            });
-        }
-
     }, [generate_file_name]);
 
     useEffect(() => {
@@ -113,14 +102,15 @@ const DataGenerator: FunctionComponent<Props> = () => {
                 ...initialData,
                 ...(mutation?.data?.dataset as any)
             });
-            form.setFieldsValue({
-                ...initialData,
-                ...(mutation?.data?.dataset as any)
-            });
-            formData.current = {
+            const dataset = mutation?.data?.dataset as any;
+            const values = {
                 ...initialData,
-                ...(mutation?.data?.dataset as any)
-            };
+                ...dataset,
+                workflow_type: dataset.technique === 'freeform' ? 
+                    WorkflowType.FREE_FORM_DATA_GENERATION : WorkflowType.CUSTOM_DATA_GENERATION
+            }
+            form.setFieldsValue(values);
+            formData.current = values;
             console.log('------> form', form.getFieldsValue());
         }
     }, [mutation.data]);