Merge pull request #1201 from mindsdb/fix/type_infer_0018

fix: add compatibility for type-infer==0.0.18
mindsdb · Dec 25, 2023 · 5409551 · 5409551
2 parents b3fd4c4 + e4e5344
commit 5409551
Show file tree

Hide file tree

Showing 5 changed files with 7 additions and 7 deletions.
diff --git a/docssrc/source/tutorials/tutorial_data_analysis/tutorial_data_analysis.ipynb b/docssrc/source/tutorials/tutorial_data_analysis/tutorial_data_analysis.ipynb
@@ -295,10 +295,10 @@
     }
    ],
    "source": [
-    "from type_infer.infer import infer_types\n",
+    "from type_infer.api import infer_types\n",
     "from type_infer.base import TypeInformation\n",
     "\n",
-    "type_information = infer_types(df, problem_definition.pct_invalid)\n",
+    "type_information = infer_types(df, config={'engine': 'rule_based', 'pct_invalid': problem_definition.pct_invalid})\n",
     "\n",
     "{field.name for field in fields(TypeInformation)}  # show the fields this dataclass has"
    ]

diff --git a/lightwood/__about__.py b/lightwood/__about__.py
@@ -1,6 +1,6 @@
 __title__ = 'lightwood'
 __package_name__ = 'lightwood'
-__version__ = '23.12.2.0'
+__version__ = '23.12.4.0'
 __description__ = "Lightwood is a toolkit for automatic machine learning model building"
 __email__ = "[email protected]"
 __author__ = 'MindsDB Inc'

diff --git a/lightwood/api/high_level.py b/lightwood/api/high_level.py
@@ -4,7 +4,7 @@
 import pandas as pd
 from lightwood.api.types import JsonAI, ProblemDefinition
 from dataprep_ml.insights import statistical_analysis
-from type_infer.infer import infer_types
+from type_infer.api import infer_types
 from lightwood.api.predictor import PredictorInterface
 from lightwood.api.json_ai import generate_json_ai
 from lightwood.helpers.codegen import code_from_json_ai as _code_from_json_ai, _module_from_code, _predictor_from_code
@@ -65,7 +65,7 @@ def json_ai_from_problem(df: pd.DataFrame, problem_definition: Union[ProblemDefi
         log.info(f'Dropping features: {problem_definition.ignore_features}')
         df = df.drop(columns=problem_definition.ignore_features)
 
-    type_information = infer_types(df, problem_definition.pct_invalid)
+    type_information = infer_types(df, config={'engine': 'rule_based', 'pct_invalid': problem_definition.pct_invalid})
     stats = statistical_analysis(
         df, type_information.dtypes, problem_definition.to_dict(), type_information.identifiers)
 

diff --git a/lightwood/encoder/text/short.py b/lightwood/encoder/text/short.py
@@ -2,7 +2,7 @@
 import torch
 from lightwood.encoder import BaseEncoder
 from lightwood.encoder.categorical import CategoricalAutoEncoder
-from type_infer.helpers import tokenize_text
+from type_infer.rule_based.helpers import tokenize_text
 from lightwood.helpers.torch import concat_vectors_and_pad, average_vectors
 import pandas as pd
 

diff --git a/tests/unit_tests/encoder/text/test_short.py b/tests/unit_tests/encoder/text/test_short.py
@@ -1,7 +1,7 @@
 import random
 import unittest
 from lightwood.encoder.text.short import ShortTextEncoder
-from type_infer.helpers import tokenize_text
+from type_infer.rule_based.helpers import tokenize_text
 import torch
 
 VOCAB = [