Fix docs

Lightning-Universe · Jan 14, 2022 · 66c8e40 · 66c8e40
1 parent cfed88c
commit 66c8e40
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/flash/tabular/classification/input.py b/flash/tabular/classification/input.py
@@ -13,8 +13,8 @@
 # limitations under the License.
 from typing import Any, Dict, List, Optional, Union
 
-from flash import DataKeys
 from flash.core.data.io.classification_input import ClassificationInputMixin
+from flash.core.data.io.input import DataKeys
 from flash.core.data.utilities.data_frame import read_csv, resolve_targets
 from flash.core.utilities.imports import _PANDAS_AVAILABLE
 from flash.tabular.input import TabularDataFrameInput

diff --git a/flash/tabular/regression/input.py b/flash/tabular/regression/input.py
@@ -15,7 +15,7 @@
 
 import numpy as np
 
-from flash import DataKeys
+from flash.core.data.io.input import DataKeys
 from flash.core.data.utilities.data_frame import read_csv
 from flash.core.utilities.imports import _PANDAS_AVAILABLE
 from flash.tabular.input import TabularDataFrameInput