#2101: Curves: Fixed fit function creation

datagrok-ai · Jul 21, 2023 · 6712528 · 6712528
1 parent d3cf02d
commit 6712528
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/libraries/statistics/src/fit/fit-curve.ts b/libraries/statistics/src/fit/fit-curve.ts
@@ -385,7 +385,7 @@ export function getOrCreateFitFunction(seriesFitFunc: string | IFitFunctionDescr
     const fitFunctionParts = seriesFitFunc.function.split('=>').map((elem) => elem.trim());
     const getInitParamsParts = seriesFitFunc.getInitialParameters.split('=>').map((elem) => elem.trim());
     const fitFunction = new Function(fitFunctionParts[0].slice(1, fitFunctionParts[0].length - 1),
-      `return ${fitFunctionParts[1]}`);
+      `${fitFunctionParts[1].includes(';') ? '' : 'return '}${fitFunctionParts[1]}`);
     const getInitParamsFunc = new Function(getInitParamsParts[0].slice(1, getInitParamsParts[0].length - 1),
       `return ${getInitParamsParts[1]}`);
     const fitFunc = new JsFunction(name, (fitFunction as (params: number[], x: number) => number),