FLAML 2.5.0__tar.gz → 2.7.0__tar.gz

{flaml-2.5.0 → flaml-2.7.0}/FLAML.egg-info/PKG-INFO

@@ -1,16 +1,17 @@
 Metadata-Version: 2.4
 Name: FLAML
-Version: 2.5.0
+Version: 2.7.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
+Author-email: Microsoft Corporation <hpo@microsoft.com>
+License: MIT
+Project-URL: Homepage, https://github.com/microsoft/FLAML
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
@@ -26,48 +27,50 @@ Provides-Extra: notebook
 Requires-Dist: jupyter; extra == "notebook"
 Provides-Extra: spark
 Requires-Dist: pyspark>=3.2.0; extra == "spark"
+Requires-Dist: pandas<3; extra == "spark"
 Requires-Dist: joblibspark>=0.5.0; extra == "spark"
 Requires-Dist: joblib<=1.3.2; extra == "spark"
 Provides-Extra: test
-Requires-Dist: numpy<2.0.0,>=1.17; python_version < "3.13" and extra == "test"
-Requires-Dist: numpy>=1.17; python_version >= "3.13" and extra == "test"
+Requires-Dist: numpy<2.0.0,>=1.17; (python_version < "3.13" and (sys_platform != "win32" or platform_machine != "ARM64")) and extra == "test"
+Requires-Dist: numpy>=1.17; (python_version >= "3.13" or (sys_platform == "win32" and platform_machine == "ARM64")) and extra == "test"
 Requires-Dist: jupyter; extra == "test"
 Requires-Dist: lightgbm>=2.3.1; extra == "test"
 Requires-Dist: xgboost<2.0.0,>=0.90; python_version < "3.11" and extra == "test"
 Requires-Dist: xgboost>=2.0.0; python_version >= "3.11" and extra == "test"
 Requires-Dist: scipy>=1.4.1; 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: pandas>=1.1.4; extra == "test"
 Requires-Dist: scikit-learn>=1.2.0; extra == "test"
-Requires-Dist: thop; extra == "test"
+Requires-Dist: thop; (sys_platform != "win32" or platform_machine != "ARM64") and extra == "test"
 Requires-Dist: pytest>=6.1.1; extra == "test"
 Requires-Dist: pytest-rerunfailures>=13.0; extra == "test"
+Requires-Dist: pytest-asyncio; (sys_platform == "win32" and platform_machine == "ARM64") and 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>=0.26; extra == "test"
+Requires-Dist: torch; (sys_platform != "win32" or platform_machine != "ARM64") and extra == "test"
+Requires-Dist: torchvision; (sys_platform != "win32" or platform_machine != "ARM64") and extra == "test"
+Requires-Dist: catboost>=0.26; (sys_platform != "win32" or platform_machine != "ARM64") and extra == "test"
 Requires-Dist: rgf-python; extra == "test"
 Requires-Dist: optuna<=3.6.1,>=2.8.0; extra == "test"
-Requires-Dist: openml; extra == "test"
+Requires-Dist: openml; (sys_platform != "win32" or platform_machine != "ARM64") and extra == "test"
 Requires-Dist: statsmodels>=0.12.2; extra == "test"
 Requires-Dist: psutil; extra == "test"
-Requires-Dist: dataclasses; extra == "test"
-Requires-Dist: transformers[torch]; extra == "test"
-Requires-Dist: datasets; extra == "test"
-Requires-Dist: evaluate; extra == "test"
+Requires-Dist: transformers[torch]; (sys_platform != "win32" or platform_machine != "ARM64") and extra == "test"
+Requires-Dist: transformers; (sys_platform == "win32" and platform_machine == "ARM64") and extra == "test"
+Requires-Dist: datasets; (sys_platform != "win32" or platform_machine != "ARM64") and extra == "test"
+Requires-Dist: evaluate; (sys_platform != "win32" or platform_machine != "ARM64") and extra == "test"
 Requires-Dist: nltk!=3.8.2; extra == "test"
 Requires-Dist: rouge_score; extra == "test"
 Requires-Dist: hcrystalball; extra == "test"
 Requires-Dist: seqeval; extra == "test"
-Requires-Dist: pytorch-forecasting; extra == "test"
+Requires-Dist: pytorch-forecasting; (sys_platform != "win32" or platform_machine != "ARM64") and extra == "test"
+Requires-Dist: cryptography==46.0.3; (sys_platform == "win32" and platform_machine == "ARM64") and extra == "test"
 Requires-Dist: mlflow-skinny<=2.22.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"
-Requires-Dist: pytorch-lightning; extra == "test"
+Requires-Dist: pytorch-lightning; (sys_platform != "win32" or platform_machine != "ARM64") and extra == "test"
 Requires-Dist: tensorboardX; extra == "test"
 Requires-Dist: requests; extra == "test"
 Requires-Dist: packaging; extra == "test"
@@ -124,17 +127,7 @@ Provides-Extra: autozero
 Requires-Dist: scikit-learn; extra == "autozero"
 Requires-Dist: pandas; extra == "autozero"
 Requires-Dist: packaging; extra == "autozero"
-Dynamic: author
-Dynamic: author-email
-Dynamic: classifier
-Dynamic: description
-Dynamic: description-content-type
-Dynamic: home-page
 Dynamic: license-file
-Dynamic: provides-extra
-Dynamic: requires-dist
-Dynamic: requires-python
-Dynamic: summary
 
 [![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)
@@ -152,7 +145,11 @@ Dynamic: summary
     <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 and 3.12 support, along with a range of new estimators, and comprehensive integration with MLflow—thanks to contributions from the Microsoft Fabric product team.
+<p align="center">
+  <b>English</b> · <a href="https://github.com/microsoft/FLAML/blob/main/README.zh.md">简体中文</a>
+</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: [AutoGen](https://microsoft.github.io/autogen/) has moved to a dedicated [GitHub repository](https://github.com/microsoft/autogen). FLAML no longer includes the `autogen` module—please use AutoGen directly.
 
@@ -212,7 +209,7 @@ automl.fit(X_train, y_train, task="classification", estimator_list=["lgbm"])
 from flaml import tune
 
 tune.run(
-    evaluation_function, config={…}, low_cost_partial_config={…}, time_budget_s=3600
+    evaluation_function, config={...}, low_cost_partial_config={...}, time_budget_s=3600
 )
 ```
 
@@ -247,7 +244,7 @@ This project welcomes contributions and suggestions. Most contributions require
 Contributor License Agreement (CLA) declaring that you have the right to, and actually do, grant us
 the rights to use your contribution. For details, visit <https://cla.opensource.microsoft.com>.
 
-If you are new to GitHub [here](https://help.github.com/categories/collaborating-with-issues-and-pull-requests/) is a detailed help source on getting involved with development on GitHub.
+If you are new to GitHub [here](https://docs.github.com/en/pull-requests/collaborating-with-pull-requests) is a detailed help source on getting involved with development on GitHub.
 
 When you submit a pull request, a CLA bot will automatically determine whether you need to provide
 a CLA and decorate the PR appropriately (e.g., status check, comment). Simply follow the instructions

{flaml-2.5.0 → flaml-2.7.0}/FLAML.egg-info/SOURCES.txt

@@ -11,6 +11,7 @@ FLAML.egg-info/top_level.txt
 flaml/__init__.py
 flaml/config.py
 flaml/ml.py
+flaml/py.typed
 flaml/version.py
 flaml/autogen/__init__.py
 flaml/autogen/code_utils.py
@@ -22,6 +23,9 @@ flaml/autogen/agentchat/assistant_agent.py
 flaml/autogen/agentchat/conversable_agent.py
 flaml/autogen/agentchat/groupchat.py
 flaml/autogen/agentchat/user_proxy_agent.py
+flaml/autogen/agentchat/contrib/math_user_proxy_agent.py
+flaml/autogen/agentchat/contrib/retrieve_assistant_agent.py
+flaml/autogen/agentchat/contrib/retrieve_user_proxy_agent.py
 flaml/autogen/extensions/__init__.py
 flaml/autogen/oai/__init__.py
 flaml/autogen/oai/completion.py

{flaml-2.5.0 → flaml-2.7.0}/FLAML.egg-info/requires.txt

@@ -62,6 +62,7 @@ ray[tune]<2.5.0,>=1.13
 
 [spark]
 pyspark>=3.2.0
+pandas<3
 joblibspark>=0.5.0
 joblib<=1.3.2
 
@@ -74,59 +75,60 @@ pyspark>=3.2.0
 jupyter
 lightgbm>=2.3.1
 scipy>=1.4.1
+pandas>=1.1.4
 scikit-learn>=1.2.0
-thop
 pytest>=6.1.1
 pytest-rerunfailures>=13.0
 coverage>=5.3
 pre-commit
-torch
-torchvision
-catboost>=0.26
 rgf-python
 optuna<=3.6.1,>=2.8.0
-openml
 statsmodels>=0.12.2
 psutil
-dataclasses
-transformers[torch]
-datasets
-evaluate
 nltk!=3.8.2
 rouge_score
 hcrystalball
 seqeval
-pytorch-forecasting
 mlflow-skinny<=2.22.1
 joblibspark>=0.5.0
 joblib<=1.3.2
 nbconvert
 nbformat
 ipykernel
-pytorch-lightning
 tensorboardX
 requests
 packaging
 dill
 
-[test:python_version < "3.10"]
-pandas<2.0.0,>=1.1.4
-
 [test:python_version < "3.11"]
 xgboost<2.0.0,>=0.90
 
-[test:python_version < "3.13"]
+[test:python_version < "3.13" and (sys_platform != "win32" or platform_machine != "ARM64")]
 numpy<2.0.0,>=1.17
 
-[test:python_version >= "3.10"]
-pandas>=1.1.4
-
 [test:python_version >= "3.11"]
 xgboost>=2.0.0
 
-[test:python_version >= "3.13"]
+[test:python_version >= "3.13" or (sys_platform == "win32" and platform_machine == "ARM64")]
 numpy>=1.17
 
+[test:sys_platform != "win32" or platform_machine != "ARM64"]
+thop
+torch
+torchvision
+catboost>=0.26
+openml
+transformers[torch]
+datasets
+evaluate
+pytorch-forecasting
+pytorch-lightning
+
+[test:sys_platform == "win32" and platform_machine == "ARM64"]
+pytest-asyncio
+transformers
+cryptography==46.0.3
+
 [ts_forecast]
 holidays
 prophet>=1.1.5

{flaml-2.5.0 → flaml-2.7.0}/PKG-INFO

@@ -1,16 +1,17 @@
 Metadata-Version: 2.4
 Name: FLAML
-Version: 2.5.0
+Version: 2.7.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
+Author-email: Microsoft Corporation <hpo@microsoft.com>
+License: MIT
+Project-URL: Homepage, https://github.com/microsoft/FLAML
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
@@ -26,48 +27,50 @@ Provides-Extra: notebook
 Requires-Dist: jupyter; extra == "notebook"
 Provides-Extra: spark
 Requires-Dist: pyspark>=3.2.0; extra == "spark"
+Requires-Dist: pandas<3; extra == "spark"
 Requires-Dist: joblibspark>=0.5.0; extra == "spark"
 Requires-Dist: joblib<=1.3.2; extra == "spark"
 Provides-Extra: test
-Requires-Dist: numpy<2.0.0,>=1.17; python_version < "3.13" and extra == "test"
-Requires-Dist: numpy>=1.17; python_version >= "3.13" and extra == "test"
+Requires-Dist: numpy<2.0.0,>=1.17; (python_version < "3.13" and (sys_platform != "win32" or platform_machine != "ARM64")) and extra == "test"
+Requires-Dist: numpy>=1.17; (python_version >= "3.13" or (sys_platform == "win32" and platform_machine == "ARM64")) and extra == "test"
 Requires-Dist: jupyter; extra == "test"
 Requires-Dist: lightgbm>=2.3.1; extra == "test"
 Requires-Dist: xgboost<2.0.0,>=0.90; python_version < "3.11" and extra == "test"
 Requires-Dist: xgboost>=2.0.0; python_version >= "3.11" and extra == "test"
 Requires-Dist: scipy>=1.4.1; 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: pandas>=1.1.4; extra == "test"
 Requires-Dist: scikit-learn>=1.2.0; extra == "test"
-Requires-Dist: thop; extra == "test"
+Requires-Dist: thop; (sys_platform != "win32" or platform_machine != "ARM64") and extra == "test"
 Requires-Dist: pytest>=6.1.1; extra == "test"
 Requires-Dist: pytest-rerunfailures>=13.0; extra == "test"
+Requires-Dist: pytest-asyncio; (sys_platform == "win32" and platform_machine == "ARM64") and 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>=0.26; extra == "test"
+Requires-Dist: torch; (sys_platform != "win32" or platform_machine != "ARM64") and extra == "test"
+Requires-Dist: torchvision; (sys_platform != "win32" or platform_machine != "ARM64") and extra == "test"
+Requires-Dist: catboost>=0.26; (sys_platform != "win32" or platform_machine != "ARM64") and extra == "test"
 Requires-Dist: rgf-python; extra == "test"
 Requires-Dist: optuna<=3.6.1,>=2.8.0; extra == "test"
-Requires-Dist: openml; extra == "test"
+Requires-Dist: openml; (sys_platform != "win32" or platform_machine != "ARM64") and extra == "test"
 Requires-Dist: statsmodels>=0.12.2; extra == "test"
 Requires-Dist: psutil; extra == "test"
-Requires-Dist: dataclasses; extra == "test"
-Requires-Dist: transformers[torch]; extra == "test"
-Requires-Dist: datasets; extra == "test"
-Requires-Dist: evaluate; extra == "test"
+Requires-Dist: transformers[torch]; (sys_platform != "win32" or platform_machine != "ARM64") and extra == "test"
+Requires-Dist: transformers; (sys_platform == "win32" and platform_machine == "ARM64") and extra == "test"
+Requires-Dist: datasets; (sys_platform != "win32" or platform_machine != "ARM64") and extra == "test"
+Requires-Dist: evaluate; (sys_platform != "win32" or platform_machine != "ARM64") and extra == "test"
 Requires-Dist: nltk!=3.8.2; extra == "test"
 Requires-Dist: rouge_score; extra == "test"
 Requires-Dist: hcrystalball; extra == "test"
 Requires-Dist: seqeval; extra == "test"
-Requires-Dist: pytorch-forecasting; extra == "test"
+Requires-Dist: pytorch-forecasting; (sys_platform != "win32" or platform_machine != "ARM64") and extra == "test"
+Requires-Dist: cryptography==46.0.3; (sys_platform == "win32" and platform_machine == "ARM64") and extra == "test"
 Requires-Dist: mlflow-skinny<=2.22.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"
-Requires-Dist: pytorch-lightning; extra == "test"
+Requires-Dist: pytorch-lightning; (sys_platform != "win32" or platform_machine != "ARM64") and extra == "test"
 Requires-Dist: tensorboardX; extra == "test"
 Requires-Dist: requests; extra == "test"
 Requires-Dist: packaging; extra == "test"
@@ -124,17 +127,7 @@ Provides-Extra: autozero
 Requires-Dist: scikit-learn; extra == "autozero"
 Requires-Dist: pandas; extra == "autozero"
 Requires-Dist: packaging; extra == "autozero"
-Dynamic: author
-Dynamic: author-email
-Dynamic: classifier
-Dynamic: description
-Dynamic: description-content-type
-Dynamic: home-page
 Dynamic: license-file
-Dynamic: provides-extra
-Dynamic: requires-dist
-Dynamic: requires-python
-Dynamic: summary
 
 [![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)
@@ -152,7 +145,11 @@ Dynamic: summary
     <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 and 3.12 support, along with a range of new estimators, and comprehensive integration with MLflow—thanks to contributions from the Microsoft Fabric product team.
+<p align="center">
+  <b>English</b> · <a href="https://github.com/microsoft/FLAML/blob/main/README.zh.md">简体中文</a>
+</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: [AutoGen](https://microsoft.github.io/autogen/) has moved to a dedicated [GitHub repository](https://github.com/microsoft/autogen). FLAML no longer includes the `autogen` module—please use AutoGen directly.
 
@@ -212,7 +209,7 @@ automl.fit(X_train, y_train, task="classification", estimator_list=["lgbm"])
 from flaml import tune
 
 tune.run(
-    evaluation_function, config={…}, low_cost_partial_config={…}, time_budget_s=3600
+    evaluation_function, config={...}, low_cost_partial_config={...}, time_budget_s=3600
 )
 ```
 
@@ -247,7 +244,7 @@ This project welcomes contributions and suggestions. Most contributions require
 Contributor License Agreement (CLA) declaring that you have the right to, and actually do, grant us
 the rights to use your contribution. For details, visit <https://cla.opensource.microsoft.com>.
 
-If you are new to GitHub [here](https://help.github.com/categories/collaborating-with-issues-and-pull-requests/) is a detailed help source on getting involved with development on GitHub.
+If you are new to GitHub [here](https://docs.github.com/en/pull-requests/collaborating-with-pull-requests) is a detailed help source on getting involved with development on GitHub.
 
 When you submit a pull request, a CLA bot will automatically determine whether you need to provide
 a CLA and decorate the PR appropriately (e.g., status check, comment). Simply follow the instructions

{flaml-2.5.0 → flaml-2.7.0}/README.md

@@ -14,7 +14,11 @@
     <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 and 3.12 support, along with a range of new estimators, and comprehensive integration with MLflow—thanks to contributions from the Microsoft Fabric product team.
+<p align="center">
+  <b>English</b> · <a href="https://github.com/microsoft/FLAML/blob/main/README.zh.md">简体中文</a>
+</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: [AutoGen](https://microsoft.github.io/autogen/) has moved to a dedicated [GitHub repository](https://github.com/microsoft/autogen). FLAML no longer includes the `autogen` module—please use AutoGen directly.
 
@@ -74,7 +78,7 @@ automl.fit(X_train, y_train, task="classification", estimator_list=["lgbm"])
 from flaml import tune
 
 tune.run(
-    evaluation_function, config={…}, low_cost_partial_config={…}, time_budget_s=3600
+    evaluation_function, config={...}, low_cost_partial_config={...}, time_budget_s=3600
 )
 ```
 
@@ -109,7 +113,7 @@ This project welcomes contributions and suggestions. Most contributions require
 Contributor License Agreement (CLA) declaring that you have the right to, and actually do, grant us
 the rights to use your contribution. For details, visit <https://cla.opensource.microsoft.com>.
 
-If you are new to GitHub [here](https://help.github.com/categories/collaborating-with-issues-and-pull-requests/) is a detailed help source on getting involved with development on GitHub.
+If you are new to GitHub [here](https://docs.github.com/en/pull-requests/collaborating-with-pull-requests) is a detailed help source on getting involved with development on GitHub.
 
 When you submit a pull request, a CLA bot will automatically determine whether you need to provide
 a CLA and decorate the PR appropriately (e.g., status check, comment). Simply follow the instructions