FLAML 2.1.2__tar.gz → 2.3.0__tar.gz

{FLAML-2.1.2 → flaml-2.3.0}/FLAML.egg-info/PKG-INFO

@@ -1,58 +1,66 @@
 Metadata-Version: 2.1
 Name: FLAML
-Version: 2.1.2
+Version: 2.3.0
 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
 Requires-Dist: NumPy>=1.17
 Provides-Extra: automl
 Requires-Dist: lightgbm>=2.3.1; extra == "automl"
-Requires-Dist: xgboost>=0.90; extra == "automl"
+Requires-Dist: xgboost<3.0.0,>=0.90; extra == "automl"
 Requires-Dist: scipy>=1.4.1; extra == "automl"
 Requires-Dist: pandas>=1.1.4; extra == "automl"
-Requires-Dist: scikit-learn>=0.24; extra == "automl"
+Requires-Dist: scikit-learn>=1.0.0; extra == "automl"
 Provides-Extra: notebook
 Requires-Dist: jupyter; extra == "notebook"
 Provides-Extra: spark
 Requires-Dist: pyspark>=3.2.0; extra == "spark"
 Requires-Dist: joblibspark>=0.5.0; extra == "spark"
+Requires-Dist: joblib<=1.3.2; extra == "spark"
 Provides-Extra: test
+Requires-Dist: jupyter; extra == "test"
 Requires-Dist: lightgbm>=2.3.1; extra == "test"
-Requires-Dist: xgboost>=0.90; 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: scikit-learn>=0.24; 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"
 Requires-Dist: coverage>=5.3; extra == "test"
 Requires-Dist: pre-commit; extra == "test"
 Requires-Dist: torch; extra == "test"
 Requires-Dist: torchvision; extra == "test"
-Requires-Dist: catboost<1.2,>=0.26; extra == "test"
+Requires-Dist: catboost<1.2,>=0.26; python_version < "3.11" and extra == "test"
+Requires-Dist: catboost>=0.26; python_version >= "3.11" and extra == "test"
 Requires-Dist: rgf-python; extra == "test"
-Requires-Dist: optuna==2.8.0; extra == "test"
+Requires-Dist: optuna<=3.6.1,>=2.8.0; extra == "test"
 Requires-Dist: openml; extra == "test"
 Requires-Dist: statsmodels>=0.12.2; extra == "test"
 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; extra == "test"
-Requires-Dist: mlflow; extra == "test"
-Requires-Dist: pyspark>=3.2.0; extra == "test"
+Requires-Dist: pytorch-forecasting<=0.10.1,>=0.9.0; python_version < "3.11" and 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"
 Requires-Dist: nbformat; extra == "test"
 Requires-Dist: ipykernel; extra == "test"
@@ -63,10 +71,12 @@ 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>=0.26; 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"
 Provides-Extra: blendsearch
-Requires-Dist: optuna==2.8.0; extra == "blendsearch"
+Requires-Dist: optuna<=3.6.1,>=2.8.0; extra == "blendsearch"
 Requires-Dist: packaging; extra == "blendsearch"
 Provides-Extra: ray
 Requires-Dist: ray[tune]~=1.13; extra == "ray"
@@ -80,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
@@ -99,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
@@ -130,7 +140,7 @@ Requires-Dist: tiktoken; extra == "retrievechat"
 Requires-Dist: sentence_transformers; extra == "retrievechat"
 Provides-Extra: synapse
 Requires-Dist: joblibspark>=0.5.0; extra == "synapse"
-Requires-Dist: optuna==2.8.0; extra == "synapse"
+Requires-Dist: optuna<=3.6.1,>=2.8.0; extra == "synapse"
 Requires-Dist: pyspark>=3.2.0; extra == "synapse"
 Provides-Extra: autozero
 Requires-Dist: scikit-learn; extra == "autozero"
@@ -140,11 +150,11 @@ 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)
-<!-- [![Join the chat at https://gitter.im/FLAMLer/community](https://badges.gitter.im/FLAMLer/community.svg)](https://gitter.im/FLAMLer/community?utm_source=badge&utm_medium=badge&utm_campaign=pr-badge&utm_content=badge) -->
 
+<!-- [![Join the chat at https://gitter.im/FLAMLer/community](https://badges.gitter.im/FLAMLer/community.svg)](https://gitter.im/FLAMLer/community?utm_source=badge&utm_medium=badge&utm_campaign=pr-badge&utm_content=badge) -->
 
 # A Fast Library for Automated Machine Learning & Tuning
 
@@ -153,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.
@@ -161,17 +173,15 @@ 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
 learning and AI operations. It automates workflow based on large language models, machine learning models, etc.
 and optimizes their performance.
 
-* FLAML enables building next-gen GPT-X applications based on multi-agent conversations with minimal effort. It simplifies the orchestration, automation and optimization of a complex GPT-X workflow. It maximizes the performance of GPT-X models and augments their weakness.
-* For common machine learning tasks like classification and regression, it quickly finds quality models for user-provided data with low computational resources. It is easy to customize or extend. Users can find their desired customizability from a smooth range.
-* It supports fast and economical automatic tuning (e.g., inference hyperparameters for foundation models, configurations in MLOps/LMOps workflows, pipelines, mathematical/statistical models, algorithms, computing experiments, software configurations), capable of handling large search space with heterogeneous evaluation cost and complex constraints/guidance/early stopping.
+- FLAML enables building next-gen GPT-X applications based on multi-agent conversations with minimal effort. It simplifies the orchestration, automation and optimization of a complex GPT-X workflow. It maximizes the performance of GPT-X models and augments their weakness.
+- For common machine learning tasks like classification and regression, it quickly finds quality models for user-provided data with low computational resources. It is easy to customize or extend. Users can find their desired customizability from a smooth range.
+- It supports fast and economical automatic tuning (e.g., inference hyperparameters for foundation models, configurations in MLOps/LMOps workflows, pipelines, mathematical/statistical models, algorithms, computing experiments, software configurations), capable of handling large search space with heterogeneous evaluation cost and complex constraints/guidance/early stopping.
 
 FLAML is powered by a series of [research studies](https://microsoft.github.io/FLAML/docs/Research/) from Microsoft Research and collaborators such as Penn State University, Stevens Institute of Technology, University of Washington, and University of Waterloo.
 
@@ -186,6 +196,7 @@ pip install flaml
 ```
 
 Minimal dependencies are installed without extra options. You can install extra options based on the feature you need. For example, use the following to install the dependencies needed by the [`autogen`](https://microsoft.github.io/autogen/) package.
+
 ```bash
 pip install "flaml[autogen]"
 ```
@@ -195,18 +206,24 @@ Each of the [`notebook examples`](https://github.com/microsoft/FLAML/tree/main/n
 
 ## Quickstart
 
-* (New) The [autogen](https://microsoft.github.io/autogen/) package enables the next-gen GPT-X applications with a generic multi-agent conversation framework.
-It offers customizable and conversable agents which integrate LLMs, tools and human.
-By automating chat among multiple capable agents, one can easily make them collectively perform tasks autonomously or with human feedback, including tasks that require using tools via code. For example,
+- (New) The [autogen](https://microsoft.github.io/autogen/) package enables the next-gen GPT-X applications with a generic multi-agent conversation framework.
+  It offers customizable and conversable agents which integrate LLMs, tools and human.
+  By automating chat among multiple capable agents, one can easily make them collectively perform tasks autonomously or with human feedback, including tasks that require using tools via code. For example,
+
 ```python
 from flaml import autogen
+
 assistant = autogen.AssistantAgent("assistant")
 user_proxy = autogen.UserProxyAgent("user_proxy")
-user_proxy.initiate_chat(assistant, message="Show me the YTD gain of 10 largest technology companies as of today.")
+user_proxy.initiate_chat(
+    assistant,
+    message="Show me the YTD gain of 10 largest technology companies as of today.",
+)
 # This initiates an automated chat between the two agents to solve the task
 ```
 
 Autogen also helps maximize the utility out of the expensive LLMs such as ChatGPT and GPT-4. It offers a drop-in replacement of `openai.Completion` or `openai.ChatCompletion` with powerful functionalites like tuning, caching, templating, filtering. For example, you can optimize generations by LLM with your own tuning data, success metrics and budgets.
+
 ```python
 # perform tuning
 config, analysis = autogen.Completion.tune(
@@ -221,30 +238,32 @@ config, analysis = autogen.Completion.tune(
 # perform inference for a test instance
 response = autogen.Completion.create(context=test_instance, **config)
 ```
-* With three lines of code, you can start using this economical and fast
-AutoML engine as a [scikit-learn style estimator](https://microsoft.github.io/FLAML/docs/Use-Cases/Task-Oriented-AutoML).
+
+- With three lines of code, you can start using this economical and fast
+  AutoML engine as a [scikit-learn style estimator](https://microsoft.github.io/FLAML/docs/Use-Cases/Task-Oriented-AutoML).
 
 ```python
 from flaml import AutoML
+
 automl = AutoML()
 automl.fit(X_train, y_train, task="classification")
 ```
 
-* You can restrict the learners and use FLAML as a fast hyperparameter tuning
-tool for XGBoost, LightGBM, Random Forest etc. or a [customized learner](https://microsoft.github.io/FLAML/docs/Use-Cases/Task-Oriented-AutoML#estimator-and-search-space).
+- You can restrict the learners and use FLAML as a fast hyperparameter tuning
+  tool for XGBoost, LightGBM, Random Forest etc. or a [customized learner](https://microsoft.github.io/FLAML/docs/Use-Cases/Task-Oriented-AutoML#estimator-and-search-space).
 
 ```python
 automl.fit(X_train, y_train, task="classification", estimator_list=["lgbm"])
 ```
 
-* You can also run generic hyperparameter tuning for a [custom function](https://microsoft.github.io/FLAML/docs/Use-Cases/Tune-User-Defined-Function).
+- You can also run generic hyperparameter tuning for a [custom function](https://microsoft.github.io/FLAML/docs/Use-Cases/Tune-User-Defined-Function).
 
 ```python
 from flaml import tune
 tune.run(evaluation_function, config={…}, low_cost_partial_config={…}, time_budget_s=3600)
 ```
 
-* [Zero-shot AutoML](https://microsoft.github.io/FLAML/docs/Use-Cases/Zero-Shot-AutoML) allows using the existing training API from lightgbm, xgboost etc. while getting the benefit of AutoML in choosing high-performance hyperparameter configurations per task.
+- [Zero-shot AutoML](https://microsoft.github.io/FLAML/docs/Use-Cases/Zero-Shot-AutoML) allows using the existing training API from lightgbm, xgboost etc. while getting the benefit of AutoML in choosing high-performance hyperparameter configurations per task.
 
 ```python
 from flaml.default import LGBMRegressor

{FLAML-2.1.2 → flaml-2.3.0}/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.1.2 → flaml-2.3.0}/FLAML.egg-info/requires.txt

@@ -7,10 +7,10 @@ termcolor
 
 [automl]
 lightgbm>=2.3.1
-xgboost>=0.90
+xgboost<3.0.0,>=0.90
 scipy>=1.4.1
 pandas>=1.1.4
-scikit-learn>=0.24
+scikit-learn>=1.0.0
 
 [autozero]
 scikit-learn
@@ -27,25 +27,32 @@ xgboost==1.3.3
 pandas==1.1.4
 
 [blendsearch]
-optuna==2.8.0
+optuna<=3.6.1,>=2.8.0
 packaging
 
 [catboost]
-catboost>=0.26
+
+[catboost:python_version < "3.11"]
+catboost<1.2,>=0.26
+
+[catboost:python_version >= "3.11"]
+catboost<=1.2.5,>=0.26
 
 [forecast]
 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
 
@@ -60,7 +67,7 @@ wolframalpha
 [nlp]
 transformers[torch]==4.26
 datasets
-nltk
+nltk<=3.8.1
 rouge_score
 seqeval
 
@@ -88,41 +95,40 @@ sentence_transformers
 [spark]
 pyspark>=3.2.0
 joblibspark>=0.5.0
+joblib<=1.3.2
 
 [synapse]
 joblibspark>=0.5.0
-optuna==2.8.0
+optuna<=3.6.1,>=2.8.0
 pyspark>=3.2.0
 
 [test]
+jupyter
 lightgbm>=2.3.1
-xgboost>=0.90
+xgboost<2.0.0,>=0.90
 scipy>=1.4.1
-pandas>=1.1.4
-scikit-learn>=0.24
+scikit-learn>=1.0.0
 thop
 pytest>=6.1.1
 coverage>=5.3
 pre-commit
 torch
 torchvision
-catboost<1.2,>=0.26
 rgf-python
-optuna==2.8.0
+optuna<=3.6.1,>=2.8.0
 openml
 statsmodels>=0.12.2
 psutil==5.8.0
 dataclasses
 transformers[torch]==4.26
 datasets
-nltk
+nltk<=3.8.1
 rouge_score
 hcrystalball==0.1.10
 seqeval
-pytorch-forecasting<=0.10.1,>=0.9.0
-mlflow
-pyspark>=3.2.0
+mlflow==2.15.1
 joblibspark>=0.5.0
+joblib<=1.3.2
 nbconvert
 nbformat
 ipykernel
@@ -133,6 +139,20 @@ 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
 
 [ts_forecast]
 holidays<0.14