From 4f505d1ff86bb051a0deabcb03ec3fbac57bad02 Mon Sep 17 00:00:00 2001
From: Walter Rafelsberger <walter@elastic.co>
Date: Fri, 5 Feb 2021 11:05:30 +0100
Subject: [PATCH] [ML] Fix types.

---
 .../public/application/components/scatterplot_matrix/index.ts | 4 +---
 .../scatterplot_matrix_vega_lite_spec.test.ts                 | 3 ++-
 2 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/x-pack/plugins/ml/public/application/components/scatterplot_matrix/index.ts b/x-pack/plugins/ml/public/application/components/scatterplot_matrix/index.ts
index c72b0eb5fd66e5..216b0d8d5e9920 100644
--- a/x-pack/plugins/ml/public/application/components/scatterplot_matrix/index.ts
+++ b/x-pack/plugins/ml/public/application/components/scatterplot_matrix/index.ts
@@ -6,6 +6,4 @@
  */
 
 export { useScatterplotFieldOptions } from './use_scatterplot_field_options';
-export { LEGEND_TYPES } from './scatterplot_matrix_vega_lite_spec';
-export { ScatterplotMatrix } from './scatterplot_matrix';
-export type { ScatterplotMatrixViewProps as ScatterplotMatrixProps } from './scatterplot_matrix_view';
+export { ScatterplotMatrix, ScatterplotMatrixProps } from './scatterplot_matrix';
diff --git a/x-pack/plugins/ml/public/application/components/scatterplot_matrix/scatterplot_matrix_vega_lite_spec.test.ts b/x-pack/plugins/ml/public/application/components/scatterplot_matrix/scatterplot_matrix_vega_lite_spec.test.ts
index 44fba189e856cf..d428b5208eb1a7 100644
--- a/x-pack/plugins/ml/public/application/components/scatterplot_matrix/scatterplot_matrix_vega_lite_spec.test.ts
+++ b/x-pack/plugins/ml/public/application/components/scatterplot_matrix/scatterplot_matrix_vega_lite_spec.test.ts
@@ -10,13 +10,14 @@ import { compile } from 'vega-lite/build-es5/vega-lite';
 
 import euiThemeLight from '@elastic/eui/dist/eui_theme_light.json';
 
+import { LEGEND_TYPES } from '../vega_chart';
+
 import {
   getColorSpec,
   getScatterplotMatrixVegaLiteSpec,
   COLOR_OUTLIER,
   COLOR_RANGE_NOMINAL,
   DEFAULT_COLOR,
-  LEGEND_TYPES,
 } from './scatterplot_matrix_vega_lite_spec';
 
 describe('getColorSpec()', () => {