[ML] Fix type

elastic · Apr 5, 2021 · b7f8427 · b7f8427
1 parent 25c1df6
commit b7f8427
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/...alytics/pages/analytics_creation/components/configuration_step/form_options_validation.ts b/...alytics/pages/analytics_creation/components/configuration_step/form_options_validation.ts
@@ -6,8 +6,8 @@
  */
 
 import { i18n } from '@kbn/i18n';
+import { estypes } from '@elastic/elasticsearch';
 import { ES_FIELD_TYPES } from '../../../../../../../../../../src/plugins/data/public';
-import { RuntimeType } from '../../../../../../../../../../src/plugins/data/common';
 import { EVENT_RATE_FIELD_ID } from '../../../../../../../common/types/fields';
 import { ANALYSIS_CONFIG_TYPE } from '../../../../common/analytics';
 import { AnalyticsJobType } from '../../../analytics_management/hooks/use_create_analytics_form/state';
@@ -18,7 +18,7 @@ export const CATEGORICAL_TYPES = new Set(['ip', 'keyword']);
 // Regression supports numeric fields. Classification supports categorical, numeric, and boolean.
 export const shouldAddAsDepVarOption = (
   fieldId: string,
-  fieldType: ES_FIELD_TYPES | RuntimeType,
+  fieldType: ES_FIELD_TYPES | estypes.RuntimeField['type'],
   jobType: AnalyticsJobType
 ) => {
   if (fieldId === EVENT_RATE_FIELD_ID) return false;

diff --git a/.../public/application/data_frame_analytics/pages/analytics_creation/hooks/use_index_data.ts b/.../public/application/data_frame_analytics/pages/analytics_creation/hooks/use_index_data.ts
@@ -13,7 +13,7 @@ import { CoreSetup } from 'src/core/public';
 
 import { IndexPattern } from '../../../../../../../../../src/plugins/data/public';
 import { isRuntimeMappings } from '../../../../../../common/util/runtime_field_utils';
-import { RuntimeMappings, RuntimeField } from '../../../../../../common/types/fields';
+import { RuntimeMappings } from '../../../../../../common/types/fields';
 import { DEFAULT_SAMPLER_SHARD_SIZE } from '../../../../../../common/constants/field_histograms';
 
 import { DataLoader } from '../../../../datavisualizer/index_based/data_loader';
@@ -44,7 +44,7 @@ interface MLEuiDataGridColumn extends EuiDataGridColumn {
 function getRuntimeFieldColumns(runtimeMappings: RuntimeMappings) {
   return Object.keys(runtimeMappings).map((id) => {
     const field = runtimeMappings[id];
-    const schema = getDataGridSchemaFromESFieldType(field.type as RuntimeField['type']);
+    const schema = getDataGridSchemaFromESFieldType(field.type as estypes.RuntimeField['type']);
     return { id, schema, isExpandable: schema !== 'boolean', isRuntimeFieldColumn: true };
   });
 }
@@ -64,7 +64,7 @@ export const useIndexData = (
       const field = indexPattern.fields.getByName(id);
       const isRuntimeFieldColumn = field?.runtimeField !== undefined;
       const schema = isRuntimeFieldColumn
-        ? getDataGridSchemaFromESFieldType(field?.type as RuntimeField['type'])
+        ? getDataGridSchemaFromESFieldType(field?.type as estypes.RuntimeField['type'])
         : getDataGridSchemaFromKibanaFieldType(field);
       return {
         id,