FLAML 2.2.0__tar.gz → 2.3.1__tar.gz

{flaml-2.2.0 → flaml-2.3.1}/FLAML.egg-info/PKG-INFO

@@ -1,14 +1,18 @@
 Metadata-Version: 2.1
 Name: FLAML
-Version: 2.2.0
+Version: 2.3.1
 Summary: A fast library for automated machine learning and tuning
 Home-page: https://github.com/microsoft/FLAML
 Author: Microsoft Corporation
 Author-email: hpo@microsoft.com
-Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
-Requires-Python: >=3.6
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 License-File: LICENSE
 License-File: NOTICE.md
@@ -30,7 +34,8 @@ Requires-Dist: jupyter; extra == "test"
 Requires-Dist: lightgbm>=2.3.1; extra == "test"
 Requires-Dist: xgboost<2.0.0,>=0.90; extra == "test"
 Requires-Dist: scipy>=1.4.1; extra == "test"
-Requires-Dist: pandas>=1.1.4; extra == "test"
+Requires-Dist: pandas<2.0.0,>=1.1.4; python_version < "3.10" and extra == "test"
+Requires-Dist: pandas>=1.1.4; python_version >= "3.10" and extra == "test"
 Requires-Dist: scikit-learn>=1.0.0; extra == "test"
 Requires-Dist: thop; extra == "test"
 Requires-Dist: pytest>=6.1.1; extra == "test"
@@ -48,13 +53,12 @@ Requires-Dist: psutil==5.8.0; extra == "test"
 Requires-Dist: dataclasses; extra == "test"
 Requires-Dist: transformers[torch]==4.26; extra == "test"
 Requires-Dist: datasets; extra == "test"
-Requires-Dist: nltk; extra == "test"
+Requires-Dist: nltk<=3.8.1; extra == "test"
 Requires-Dist: rouge_score; extra == "test"
 Requires-Dist: hcrystalball==0.1.10; extra == "test"
 Requires-Dist: seqeval; extra == "test"
 Requires-Dist: pytorch-forecasting<=0.10.1,>=0.9.0; python_version < "3.11" and extra == "test"
-Requires-Dist: mlflow; extra == "test"
-Requires-Dist: pyspark>=3.2.0; extra == "test"
+Requires-Dist: mlflow==2.15.1; extra == "test"
 Requires-Dist: joblibspark>=0.5.0; extra == "test"
 Requires-Dist: joblib<=1.3.2; extra == "test"
 Requires-Dist: nbconvert; extra == "test"
@@ -67,6 +71,7 @@ Requires-Dist: packaging; extra == "test"
 Requires-Dist: pydantic==1.10.9; extra == "test"
 Requires-Dist: sympy; extra == "test"
 Requires-Dist: wolframalpha; extra == "test"
+Requires-Dist: dill; extra == "test"
 Provides-Extra: catboost
 Requires-Dist: catboost<1.2,>=0.26; python_version < "3.11" and extra == "catboost"
 Requires-Dist: catboost<=1.2.5,>=0.26; python_version >= "3.11" and extra == "catboost"
@@ -85,13 +90,13 @@ Requires-Dist: scikit-learn; extra == "vw"
 Provides-Extra: hf
 Requires-Dist: transformers[torch]==4.26; extra == "hf"
 Requires-Dist: datasets; extra == "hf"
-Requires-Dist: nltk; extra == "hf"
+Requires-Dist: nltk<=3.8.1; extra == "hf"
 Requires-Dist: rouge_score; extra == "hf"
 Requires-Dist: seqeval; extra == "hf"
 Provides-Extra: nlp
 Requires-Dist: transformers[torch]==4.26; extra == "nlp"
 Requires-Dist: datasets; extra == "nlp"
-Requires-Dist: nltk; extra == "nlp"
+Requires-Dist: nltk<=3.8.1; extra == "nlp"
 Requires-Dist: rouge_score; extra == "nlp"
 Requires-Dist: seqeval; extra == "nlp"
 Provides-Extra: ts-forecast
@@ -104,7 +109,7 @@ Requires-Dist: holidays<0.14; extra == "forecast"
 Requires-Dist: prophet>=1.0.1; extra == "forecast"
 Requires-Dist: statsmodels>=0.12.2; extra == "forecast"
 Requires-Dist: hcrystalball==0.1.10; extra == "forecast"
-Requires-Dist: pytorch-forecasting>=0.9.0; extra == "forecast"
+Requires-Dist: pytorch-forecasting>=0.9.0; python_version < "3.11" and extra == "forecast"
 Requires-Dist: pytorch-lightning==1.9.0; extra == "forecast"
 Requires-Dist: tensorboardX==2.6; extra == "forecast"
 Provides-Extra: benchmark
@@ -145,7 +150,7 @@ Requires-Dist: packaging; extra == "autozero"
 [![PyPI version](https://badge.fury.io/py/FLAML.svg)](https://badge.fury.io/py/FLAML)
 ![Conda version](https://img.shields.io/conda/vn/conda-forge/flaml)
 [![Build](https://github.com/microsoft/FLAML/actions/workflows/python-package.yml/badge.svg)](https://github.com/microsoft/FLAML/actions/workflows/python-package.yml)
-![Python Version](https://img.shields.io/badge/3.8%20%7C%203.9%20%7C%203.10-blue)
+[![PyPI - Python Version](https://img.shields.io/pypi/pyversions/FLAML)](https://pypi.org/project/FLAML/)
 [![Downloads](https://pepy.tech/badge/flaml)](https://pepy.tech/project/flaml)
 [![](https://img.shields.io/discord/1025786666260111483?logo=discord&style=flat)](https://discord.gg/Cppx2vSPVP)
 
@@ -158,6 +163,8 @@ Requires-Dist: packaging; extra == "autozero"
     <br>
 </p>
 
+:fire: FLAML supports AutoML and Hyperparameter Tuning in [Microsoft Fabric Data Science](https://learn.microsoft.com/en-us/fabric/data-science/automated-machine-learning-fabric). In addition, we've introduced Python 3.11 support, along with a range of new estimators, and comprehensive integration with MLflow—thanks to contributions from the Microsoft Fabric product team.
+
 :fire: Heads-up: We have migrated [AutoGen](https://microsoft.github.io/autogen/) into a dedicated [github repository](https://github.com/microsoft/autogen). Alongside this move, we have also launched a dedicated [Discord](https://discord.gg/pAbnFJrkgZ) server and a [website](https://microsoft.github.io/autogen/) for comprehensive documentation.
 
 :fire: The automated multi-agent chat framework in [AutoGen](https://microsoft.github.io/autogen/) is in preview from v2.0.0.
@@ -166,8 +173,6 @@ Requires-Dist: packaging; extra == "autozero"
 
 :fire: [autogen](https://microsoft.github.io/autogen/) is released with support for ChatGPT and GPT-4, based on [Cost-Effective Hyperparameter Optimization for Large Language Model Generation Inference](https://arxiv.org/abs/2303.04673).
 
-:fire: FLAML supports Code-First AutoML & Tuning – Private Preview in [Microsoft Fabric Data Science](https://learn.microsoft.com/en-us/fabric/data-science/).
-
 ## What is FLAML
 
 FLAML is a lightweight Python library for efficient automation of machine
@@ -298,3 +303,9 @@ provided by the bot. You will only need to do this once across all repos using o
 This project has adopted the [Microsoft Open Source Code of Conduct](https://opensource.microsoft.com/codeofconduct/).
 For more information see the [Code of Conduct FAQ](https://opensource.microsoft.com/codeofconduct/faq/) or
 contact [opencode@microsoft.com](mailto:opencode@microsoft.com) with any additional questions or comments.
+
+## Contributors Wall
+
+<a href="https://github.com/microsoft/flaml/graphs/contributors">
+  <img src="https://contrib.rocks/image?repo=microsoft/flaml&max=204" />
+</a>

{flaml-2.2.0 → flaml-2.3.1}/FLAML.egg-info/SOURCES.txt

@@ -44,7 +44,6 @@ flaml/automl/nlp/huggingface/trainer.py
 flaml/automl/nlp/huggingface/training_args.py
 flaml/automl/nlp/huggingface/utils.py
 flaml/automl/spark/__init__.py
-flaml/automl/spark/configs.py
 flaml/automl/spark/metrics.py
 flaml/automl/spark/utils.py
 flaml/automl/task/__init__.py
@@ -55,6 +54,7 @@ flaml/automl/task/time_series_task.py
 flaml/automl/time_series/__init__.py
 flaml/automl/time_series/feature.py
 flaml/automl/time_series/sklearn.py
+flaml/automl/time_series/tcn.py
 flaml/automl/time_series/tft.py
 flaml/automl/time_series/ts_data.py
 flaml/automl/time_series/ts_model.py
@@ -82,6 +82,8 @@ flaml/default/xgb_limitdepth/regression.json
 flaml/default/xgboost/binary.json
 flaml/default/xgboost/multiclass.json
 flaml/default/xgboost/regression.json
+flaml/fabric/__init__.py
+flaml/fabric/mlflow.py
 flaml/onlineml/__init__.py
 flaml/onlineml/autovw.py
 flaml/onlineml/trial.py

{flaml-2.2.0 → flaml-2.3.1}/FLAML.egg-info/requires.txt

@@ -43,14 +43,16 @@ holidays<0.14
 prophet>=1.0.1
 statsmodels>=0.12.2
 hcrystalball==0.1.10
-pytorch-forecasting>=0.9.0
 pytorch-lightning==1.9.0
 tensorboardX==2.6
 
+[forecast:python_version < "3.11"]
+pytorch-forecasting>=0.9.0
+
 [hf]
 transformers[torch]==4.26
 datasets
-nltk
+nltk<=3.8.1
 rouge_score
 seqeval
 
@@ -65,7 +67,7 @@ wolframalpha
 [nlp]
 transformers[torch]==4.26
 datasets
-nltk
+nltk<=3.8.1
 rouge_score
 seqeval
 
@@ -105,7 +107,6 @@ jupyter
 lightgbm>=2.3.1
 xgboost<2.0.0,>=0.90
 scipy>=1.4.1
-pandas>=1.1.4
 scikit-learn>=1.0.0
 thop
 pytest>=6.1.1
@@ -121,12 +122,11 @@ psutil==5.8.0
 dataclasses
 transformers[torch]==4.26
 datasets
-nltk
+nltk<=3.8.1
 rouge_score
 hcrystalball==0.1.10
 seqeval
-mlflow
-pyspark>=3.2.0
+mlflow==2.15.1
 joblibspark>=0.5.0
 joblib<=1.3.2
 nbconvert
@@ -139,11 +139,18 @@ packaging
 pydantic==1.10.9
 sympy
 wolframalpha
+dill
+
+[test:python_version < "3.10"]
+pandas<2.0.0,>=1.1.4
 
 [test:python_version < "3.11"]
 catboost<1.2,>=0.26
 pytorch-forecasting<=0.10.1,>=0.9.0
 
+[test:python_version >= "3.10"]
+pandas>=1.1.4
+
 [test:python_version >= "3.11"]
 catboost>=0.26
 

{flaml-2.2.0 → flaml-2.3.1}/PKG-INFO

@@ -1,14 +1,18 @@
 Metadata-Version: 2.1
 Name: FLAML
-Version: 2.2.0
+Version: 2.3.1
 Summary: A fast library for automated machine learning and tuning
 Home-page: https://github.com/microsoft/FLAML
 Author: Microsoft Corporation
 Author-email: hpo@microsoft.com
-Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
-Requires-Python: >=3.6
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 License-File: LICENSE
 License-File: NOTICE.md
@@ -30,7 +34,8 @@ Requires-Dist: jupyter; extra == "test"
 Requires-Dist: lightgbm>=2.3.1; extra == "test"
 Requires-Dist: xgboost<2.0.0,>=0.90; extra == "test"
 Requires-Dist: scipy>=1.4.1; extra == "test"
-Requires-Dist: pandas>=1.1.4; extra == "test"
+Requires-Dist: pandas<2.0.0,>=1.1.4; python_version < "3.10" and extra == "test"
+Requires-Dist: pandas>=1.1.4; python_version >= "3.10" and extra == "test"
 Requires-Dist: scikit-learn>=1.0.0; extra == "test"
 Requires-Dist: thop; extra == "test"
 Requires-Dist: pytest>=6.1.1; extra == "test"
@@ -48,13 +53,12 @@ Requires-Dist: psutil==5.8.0; extra == "test"
 Requires-Dist: dataclasses; extra == "test"
 Requires-Dist: transformers[torch]==4.26; extra == "test"
 Requires-Dist: datasets; extra == "test"
-Requires-Dist: nltk; extra == "test"
+Requires-Dist: nltk<=3.8.1; extra == "test"
 Requires-Dist: rouge_score; extra == "test"
 Requires-Dist: hcrystalball==0.1.10; extra == "test"
 Requires-Dist: seqeval; extra == "test"
 Requires-Dist: pytorch-forecasting<=0.10.1,>=0.9.0; python_version < "3.11" and extra == "test"
-Requires-Dist: mlflow; extra == "test"
-Requires-Dist: pyspark>=3.2.0; extra == "test"
+Requires-Dist: mlflow==2.15.1; extra == "test"
 Requires-Dist: joblibspark>=0.5.0; extra == "test"
 Requires-Dist: joblib<=1.3.2; extra == "test"
 Requires-Dist: nbconvert; extra == "test"
@@ -67,6 +71,7 @@ Requires-Dist: packaging; extra == "test"
 Requires-Dist: pydantic==1.10.9; extra == "test"
 Requires-Dist: sympy; extra == "test"
 Requires-Dist: wolframalpha; extra == "test"
+Requires-Dist: dill; extra == "test"
 Provides-Extra: catboost
 Requires-Dist: catboost<1.2,>=0.26; python_version < "3.11" and extra == "catboost"
 Requires-Dist: catboost<=1.2.5,>=0.26; python_version >= "3.11" and extra == "catboost"
@@ -85,13 +90,13 @@ Requires-Dist: scikit-learn; extra == "vw"
 Provides-Extra: hf
 Requires-Dist: transformers[torch]==4.26; extra == "hf"
 Requires-Dist: datasets; extra == "hf"
-Requires-Dist: nltk; extra == "hf"
+Requires-Dist: nltk<=3.8.1; extra == "hf"
 Requires-Dist: rouge_score; extra == "hf"
 Requires-Dist: seqeval; extra == "hf"
 Provides-Extra: nlp
 Requires-Dist: transformers[torch]==4.26; extra == "nlp"
 Requires-Dist: datasets; extra == "nlp"
-Requires-Dist: nltk; extra == "nlp"
+Requires-Dist: nltk<=3.8.1; extra == "nlp"
 Requires-Dist: rouge_score; extra == "nlp"
 Requires-Dist: seqeval; extra == "nlp"
 Provides-Extra: ts-forecast
@@ -104,7 +109,7 @@ Requires-Dist: holidays<0.14; extra == "forecast"
 Requires-Dist: prophet>=1.0.1; extra == "forecast"
 Requires-Dist: statsmodels>=0.12.2; extra == "forecast"
 Requires-Dist: hcrystalball==0.1.10; extra == "forecast"
-Requires-Dist: pytorch-forecasting>=0.9.0; extra == "forecast"
+Requires-Dist: pytorch-forecasting>=0.9.0; python_version < "3.11" and extra == "forecast"
 Requires-Dist: pytorch-lightning==1.9.0; extra == "forecast"
 Requires-Dist: tensorboardX==2.6; extra == "forecast"
 Provides-Extra: benchmark
@@ -145,7 +150,7 @@ Requires-Dist: packaging; extra == "autozero"
 [![PyPI version](https://badge.fury.io/py/FLAML.svg)](https://badge.fury.io/py/FLAML)
 ![Conda version](https://img.shields.io/conda/vn/conda-forge/flaml)
 [![Build](https://github.com/microsoft/FLAML/actions/workflows/python-package.yml/badge.svg)](https://github.com/microsoft/FLAML/actions/workflows/python-package.yml)
-![Python Version](https://img.shields.io/badge/3.8%20%7C%203.9%20%7C%203.10-blue)
+[![PyPI - Python Version](https://img.shields.io/pypi/pyversions/FLAML)](https://pypi.org/project/FLAML/)
 [![Downloads](https://pepy.tech/badge/flaml)](https://pepy.tech/project/flaml)
 [![](https://img.shields.io/discord/1025786666260111483?logo=discord&style=flat)](https://discord.gg/Cppx2vSPVP)
 
@@ -158,6 +163,8 @@ Requires-Dist: packaging; extra == "autozero"
     <br>
 </p>
 
+:fire: FLAML supports AutoML and Hyperparameter Tuning in [Microsoft Fabric Data Science](https://learn.microsoft.com/en-us/fabric/data-science/automated-machine-learning-fabric). In addition, we've introduced Python 3.11 support, along with a range of new estimators, and comprehensive integration with MLflow—thanks to contributions from the Microsoft Fabric product team.
+
 :fire: Heads-up: We have migrated [AutoGen](https://microsoft.github.io/autogen/) into a dedicated [github repository](https://github.com/microsoft/autogen). Alongside this move, we have also launched a dedicated [Discord](https://discord.gg/pAbnFJrkgZ) server and a [website](https://microsoft.github.io/autogen/) for comprehensive documentation.
 
 :fire: The automated multi-agent chat framework in [AutoGen](https://microsoft.github.io/autogen/) is in preview from v2.0.0.
@@ -166,8 +173,6 @@ Requires-Dist: packaging; extra == "autozero"
 
 :fire: [autogen](https://microsoft.github.io/autogen/) is released with support for ChatGPT and GPT-4, based on [Cost-Effective Hyperparameter Optimization for Large Language Model Generation Inference](https://arxiv.org/abs/2303.04673).
 
-:fire: FLAML supports Code-First AutoML & Tuning – Private Preview in [Microsoft Fabric Data Science](https://learn.microsoft.com/en-us/fabric/data-science/).
-
 ## What is FLAML
 
 FLAML is a lightweight Python library for efficient automation of machine
@@ -298,3 +303,9 @@ provided by the bot. You will only need to do this once across all repos using o
 This project has adopted the [Microsoft Open Source Code of Conduct](https://opensource.microsoft.com/codeofconduct/).
 For more information see the [Code of Conduct FAQ](https://opensource.microsoft.com/codeofconduct/faq/) or
 contact [opencode@microsoft.com](mailto:opencode@microsoft.com) with any additional questions or comments.
+
+## Contributors Wall
+
+<a href="https://github.com/microsoft/flaml/graphs/contributors">
+  <img src="https://contrib.rocks/image?repo=microsoft/flaml&max=204" />
+</a>

{flaml-2.2.0 → flaml-2.3.1}/README.md

@@ -1,7 +1,7 @@
 [![PyPI version](https://badge.fury.io/py/FLAML.svg)](https://badge.fury.io/py/FLAML)
 ![Conda version](https://img.shields.io/conda/vn/conda-forge/flaml)
 [![Build](https://github.com/microsoft/FLAML/actions/workflows/python-package.yml/badge.svg)](https://github.com/microsoft/FLAML/actions/workflows/python-package.yml)
-![Python Version](https://img.shields.io/badge/3.8%20%7C%203.9%20%7C%203.10-blue)
+[![PyPI - Python Version](https://img.shields.io/pypi/pyversions/FLAML)](https://pypi.org/project/FLAML/)
 [![Downloads](https://pepy.tech/badge/flaml)](https://pepy.tech/project/flaml)
 [![](https://img.shields.io/discord/1025786666260111483?logo=discord&style=flat)](https://discord.gg/Cppx2vSPVP)
 
@@ -14,6 +14,8 @@
     <br>
 </p>
 
+:fire: FLAML supports AutoML and Hyperparameter Tuning in [Microsoft Fabric Data Science](https://learn.microsoft.com/en-us/fabric/data-science/automated-machine-learning-fabric). In addition, we've introduced Python 3.11 support, along with a range of new estimators, and comprehensive integration with MLflow—thanks to contributions from the Microsoft Fabric product team.
+
 :fire: Heads-up: We have migrated [AutoGen](https://microsoft.github.io/autogen/) into a dedicated [github repository](https://github.com/microsoft/autogen). Alongside this move, we have also launched a dedicated [Discord](https://discord.gg/pAbnFJrkgZ) server and a [website](https://microsoft.github.io/autogen/) for comprehensive documentation.
 
 :fire: The automated multi-agent chat framework in [AutoGen](https://microsoft.github.io/autogen/) is in preview from v2.0.0.
@@ -22,8 +24,6 @@
 
 :fire: [autogen](https://microsoft.github.io/autogen/) is released with support for ChatGPT and GPT-4, based on [Cost-Effective Hyperparameter Optimization for Large Language Model Generation Inference](https://arxiv.org/abs/2303.04673).
 
-:fire: FLAML supports Code-First AutoML & Tuning – Private Preview in [Microsoft Fabric Data Science](https://learn.microsoft.com/en-us/fabric/data-science/).
-
 ## What is FLAML
 
 FLAML is a lightweight Python library for efficient automation of machine
@@ -154,3 +154,9 @@ provided by the bot. You will only need to do this once across all repos using o
 This project has adopted the [Microsoft Open Source Code of Conduct](https://opensource.microsoft.com/codeofconduct/).
 For more information see the [Code of Conduct FAQ](https://opensource.microsoft.com/codeofconduct/faq/) or
 contact [opencode@microsoft.com](mailto:opencode@microsoft.com) with any additional questions or comments.
+
+## Contributors Wall
+
+<a href="https://github.com/microsoft/flaml/graphs/contributors">
+  <img src="https://contrib.rocks/image?repo=microsoft/flaml&max=204" />
+</a>

{flaml-2.2.0 → flaml-2.3.1}/flaml/__init__.py

@@ -1,6 +1,11 @@
 import logging
 
-from flaml.automl import AutoML, logger_formatter
+try:
+    from flaml.automl import AutoML, logger_formatter
+
+    has_automl = True
+except ImportError:
+    has_automl = False
 from flaml.onlineml.autovw import AutoVW
 from flaml.tune.searcher import CFO, FLOW2, BlendSearch, BlendSearchTuner, RandomSearch
 from flaml.version import __version__
@@ -8,3 +13,6 @@ from flaml.version import __version__
 # Set the root logger.
 logger = logging.getLogger(__name__)
 logger.setLevel(logging.INFO)
+
+if not has_automl:
+    logger.warning("flaml.automl is not available. Please install flaml[automl] to enable AutoML functionalities.")

{flaml-2.2.0 → flaml-2.3.1}/flaml/autogen/code_utils.py

@@ -125,7 +125,7 @@ def improve_function(file_name, func_name, objective, **config):
     """(work in progress) Improve the function to achieve the objective."""
     params = {**_IMPROVE_FUNCTION_CONFIG, **config}
     # read the entire file into a str
-    with open(file_name, "r") as f:
+    with open(file_name) as f:
         file_string = f.read()
     response = oai.Completion.create(
         {"func_name": func_name, "objective": objective, "file_string": file_string}, **params
@@ -158,7 +158,7 @@ def improve_code(files, objective, suggest_only=True, **config):
     code = ""
     for file_name in files:
         # read the entire file into a string
-        with open(file_name, "r") as f:
+        with open(file_name) as f:
             file_string = f.read()
         code += f"""{file_name}:
 {file_string}

{flaml-2.2.0 → flaml-2.3.1}/flaml/autogen/math_utils.py

@@ -130,7 +130,7 @@ def _fix_a_slash_b(string: str) -> str:
     try:
         a = int(a_str)
         b = int(b_str)
-        assert string == "{}/{}".format(a, b)
+        assert string == f"{a}/{b}"
         new_string = "\\frac{" + str(a) + "}{" + str(b) + "}"
         return new_string
     except Exception:

{flaml-2.2.0 → flaml-2.3.1}/flaml/autogen/retrieve_utils.py

@@ -126,7 +126,7 @@ def split_files_to_chunks(
     """Split a list of files into chunks of max_tokens."""
     chunks = []
     for file in files:
-        with open(file, "r") as f:
+        with open(file) as f:
             text = f.read()
         chunks += split_text_to_chunks(text, max_tokens, chunk_mode, must_break_at_empty_line)
     return chunks

flaml-2.3.1/flaml/automl/__init__.py

@@ -0,0 +1,9 @@
+from flaml.automl.logger import logger_formatter
+
+try:
+    from flaml.automl.automl import AutoML, size
+    from flaml.automl.state import AutoMLState, SearchState
+
+    __all__ = ["AutoML", "AutoMLState", "SearchState", "logger_formatter", "size"]
+except ImportError:
+    __all__ = ["logger_formatter"]