[ML] fix typing issue

elastic · Mar 22, 2020 · a46aa53 · a46aa53
1 parent c67dc3a
commit a46aa53
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/x-pack/plugins/ml/server/models/data_recognizer/data_recognizer.ts b/x-pack/plugins/ml/server/models/data_recognizer/data_recognizer.ts
@@ -380,7 +380,7 @@ export class DataRecognizer {
     end: number | undefined,
     jobOverrides: JobOverride[],
     datafeedOverrides: DatafeedOverride[],
-    estimateModelMemory: boolean | undefined
+    estimateModelMemory?: boolean
   ) {
     // load the config from disk
     const moduleConfig = await this.getModule(moduleId, jobPrefix);

diff --git a/x-pack/plugins/ml/server/shared_services/providers/modules.ts b/x-pack/plugins/ml/server/shared_services/providers/modules.ts
@@ -32,7 +32,8 @@ export interface ModulesProvider {
       start: number,
       end: number,
       jobOverrides: JobOverride[],
-      datafeedOverrides: DatafeedOverride[]
+      datafeedOverrides: DatafeedOverride[],
+      estimateModelMemory?: boolean
     ): Promise<DataRecognizerConfigResponse>;
   };
 }
@@ -65,7 +66,8 @@ export function getModulesProvider(isFullLicense: LicenseCheck): ModulesProvider
           start: number,
           end: number,
           jobOverrides: JobOverride[],
-          datafeedOverrides: DatafeedOverride[]
+          datafeedOverrides: DatafeedOverride[],
+          estimateModelMemory?: boolean
         ) {
           const dr = dataRecognizerFactory(callAsCurrentUser, savedObjectsClient);
           return dr.setupModuleItems(
@@ -79,7 +81,8 @@ export function getModulesProvider(isFullLicense: LicenseCheck): ModulesProvider
             start,
             end,
             jobOverrides,
-            datafeedOverrides
+            datafeedOverrides,
+            estimateModelMemory
           );
         },
       };