FLAML 2.1.1__tar.gz → 2.2.0__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (119) hide show
  1. {FLAML-2.1.1 → flaml-2.2.0}/FLAML.egg-info/PKG-INFO +137 -20
  2. {FLAML-2.1.1 → flaml-2.2.0}/FLAML.egg-info/SOURCES.txt +2 -0
  3. {FLAML-2.1.1 → flaml-2.2.0}/FLAML.egg-info/requires.txt +24 -14
  4. flaml-2.2.0/NOTICE.md +295 -0
  5. {FLAML-2.1.1 → flaml-2.2.0}/PKG-INFO +137 -20
  6. {FLAML-2.1.1 → flaml-2.2.0}/README.md +25 -16
  7. {FLAML-2.1.1 → flaml-2.2.0}/flaml/__init__.py +2 -2
  8. {FLAML-2.1.1 → flaml-2.2.0}/flaml/autogen/agentchat/agent.py +2 -2
  9. {FLAML-2.1.1 → flaml-2.2.0}/flaml/autogen/agentchat/assistant_agent.py +20 -19
  10. {FLAML-2.1.1 → flaml-2.2.0}/flaml/autogen/agentchat/conversable_agent.py +4 -2
  11. {FLAML-2.1.1 → flaml-2.2.0}/flaml/autogen/agentchat/groupchat.py +2 -1
  12. {FLAML-2.1.1 → flaml-2.2.0}/flaml/autogen/agentchat/user_proxy_agent.py +2 -1
  13. {FLAML-2.1.1 → flaml-2.2.0}/flaml/autogen/code_utils.py +6 -5
  14. {FLAML-2.1.1 → flaml-2.2.0}/flaml/autogen/math_utils.py +2 -1
  15. {FLAML-2.1.1 → flaml-2.2.0}/flaml/autogen/oai/__init__.py +4 -4
  16. {FLAML-2.1.1 → flaml-2.2.0}/flaml/autogen/oai/completion.py +15 -12
  17. {FLAML-2.1.1 → flaml-2.2.0}/flaml/autogen/oai/openai_utils.py +2 -2
  18. {FLAML-2.1.1 → flaml-2.2.0}/flaml/autogen/retrieve_utils.py +7 -6
  19. {FLAML-2.1.1 → flaml-2.2.0}/flaml/automl/__init__.py +1 -1
  20. {FLAML-2.1.1 → flaml-2.2.0}/flaml/automl/automl.py +27 -26
  21. flaml-2.2.0/flaml/automl/contrib/__init__.py +1 -0
  22. flaml-2.2.0/flaml/automl/contrib/histgb.py +75 -0
  23. {FLAML-2.1.1 → flaml-2.2.0}/flaml/automl/data.py +11 -7
  24. {FLAML-2.1.1 → flaml-2.2.0}/flaml/automl/ml.py +24 -19
  25. {FLAML-2.1.1 → flaml-2.2.0}/flaml/automl/model.py +42 -28
  26. {FLAML-2.1.1 → flaml-2.2.0}/flaml/automl/nlp/huggingface/data_collator.py +7 -5
  27. {FLAML-2.1.1 → flaml-2.2.0}/flaml/automl/nlp/huggingface/training_args.py +2 -1
  28. {FLAML-2.1.1 → flaml-2.2.0}/flaml/automl/nlp/huggingface/utils.py +14 -9
  29. {FLAML-2.1.1 → flaml-2.2.0}/flaml/automl/nlp/utils.py +5 -4
  30. {FLAML-2.1.1 → flaml-2.2.0}/flaml/automl/spark/__init__.py +3 -1
  31. {FLAML-2.1.1 → flaml-2.2.0}/flaml/automl/spark/metrics.py +5 -3
  32. {FLAML-2.1.1 → flaml-2.2.0}/flaml/automl/spark/utils.py +8 -6
  33. {FLAML-2.1.1 → flaml-2.2.0}/flaml/automl/state.py +4 -2
  34. {FLAML-2.1.1 → flaml-2.2.0}/flaml/automl/task/factory.py +2 -1
  35. {FLAML-2.1.1 → flaml-2.2.0}/flaml/automl/task/generic_task.py +27 -24
  36. {FLAML-2.1.1 → flaml-2.2.0}/flaml/automl/task/task.py +2 -0
  37. {FLAML-2.1.1 → flaml-2.2.0}/flaml/automl/task/time_series_task.py +17 -18
  38. {FLAML-2.1.1 → flaml-2.2.0}/flaml/automl/time_series/__init__.py +9 -10
  39. {FLAML-2.1.1 → flaml-2.2.0}/flaml/automl/time_series/tft.py +1 -0
  40. {FLAML-2.1.1 → flaml-2.2.0}/flaml/automl/time_series/ts_data.py +10 -8
  41. {FLAML-2.1.1 → flaml-2.2.0}/flaml/automl/time_series/ts_model.py +15 -16
  42. {FLAML-2.1.1 → flaml-2.2.0}/flaml/default/estimator.py +2 -0
  43. {FLAML-2.1.1 → flaml-2.2.0}/flaml/default/greedy.py +2 -2
  44. {FLAML-2.1.1 → flaml-2.2.0}/flaml/default/portfolio.py +6 -4
  45. {FLAML-2.1.1 → flaml-2.2.0}/flaml/default/regret.py +1 -0
  46. {FLAML-2.1.1 → flaml-2.2.0}/flaml/default/suggest.py +6 -4
  47. {FLAML-2.1.1 → flaml-2.2.0}/flaml/ml.py +0 -1
  48. {FLAML-2.1.1 → flaml-2.2.0}/flaml/onlineml/autovw.py +6 -5
  49. {FLAML-2.1.1 → flaml-2.2.0}/flaml/onlineml/trial.py +7 -5
  50. {FLAML-2.1.1 → flaml-2.2.0}/flaml/onlineml/trial_runner.py +4 -3
  51. {FLAML-2.1.1 → flaml-2.2.0}/flaml/tune/__init__.py +17 -18
  52. {FLAML-2.1.1 → flaml-2.2.0}/flaml/tune/analysis.py +3 -1
  53. {FLAML-2.1.1 → flaml-2.2.0}/flaml/tune/sample.py +1 -0
  54. {FLAML-2.1.1 → flaml-2.2.0}/flaml/tune/scheduler/online_scheduler.py +5 -2
  55. {FLAML-2.1.1 → flaml-2.2.0}/flaml/tune/searcher/blendsearch.py +24 -23
  56. {FLAML-2.1.1 → flaml-2.2.0}/flaml/tune/searcher/cfo_cat.py +1 -1
  57. {FLAML-2.1.1 → flaml-2.2.0}/flaml/tune/searcher/flow2.py +9 -6
  58. {FLAML-2.1.1 → flaml-2.2.0}/flaml/tune/searcher/online_searcher.py +8 -6
  59. {FLAML-2.1.1 → flaml-2.2.0}/flaml/tune/searcher/search_thread.py +5 -3
  60. {FLAML-2.1.1 → flaml-2.2.0}/flaml/tune/searcher/suggestion.py +23 -13
  61. {FLAML-2.1.1 → flaml-2.2.0}/flaml/tune/searcher/variant_generator.py +3 -1
  62. {FLAML-2.1.1 → flaml-2.2.0}/flaml/tune/space.py +4 -3
  63. {FLAML-2.1.1 → flaml-2.2.0}/flaml/tune/spark/utils.py +7 -2
  64. {FLAML-2.1.1 → flaml-2.2.0}/flaml/tune/trial.py +2 -2
  65. {FLAML-2.1.1 → flaml-2.2.0}/flaml/tune/tune.py +17 -11
  66. flaml-2.2.0/flaml/version.py +1 -0
  67. {FLAML-2.1.1 → flaml-2.2.0}/pyproject.toml +1 -0
  68. {FLAML-2.1.1 → flaml-2.2.0}/setup.py +20 -15
  69. {FLAML-2.1.1 → flaml-2.2.0}/test/test_autovw.py +23 -18
  70. {FLAML-2.1.1 → flaml-2.2.0}/test/test_conda_distribution.py +4 -2
  71. {FLAML-2.1.1 → flaml-2.2.0}/test/test_gpu.py +9 -6
  72. {FLAML-2.1.1 → flaml-2.2.0}/test/test_model.py +15 -7
  73. {FLAML-2.1.1 → flaml-2.2.0}/test/test_version.py +1 -0
  74. FLAML-2.1.1/NOTICE.md +0 -290
  75. FLAML-2.1.1/flaml/version.py +0 -1
  76. {FLAML-2.1.1 → flaml-2.2.0}/FLAML.egg-info/dependency_links.txt +0 -0
  77. {FLAML-2.1.1 → flaml-2.2.0}/FLAML.egg-info/top_level.txt +0 -0
  78. {FLAML-2.1.1 → flaml-2.2.0}/LICENSE +0 -0
  79. {FLAML-2.1.1 → flaml-2.2.0}/flaml/autogen/__init__.py +1 -1
  80. {FLAML-2.1.1 → flaml-2.2.0}/flaml/autogen/agentchat/__init__.py +2 -2
  81. {FLAML-2.1.1 → flaml-2.2.0}/flaml/autogen/extensions/__init__.py +0 -0
  82. {FLAML-2.1.1 → flaml-2.2.0}/flaml/automl/logger.py +0 -0
  83. {FLAML-2.1.1 → flaml-2.2.0}/flaml/automl/nlp/__init__.py +0 -0
  84. {FLAML-2.1.1 → flaml-2.2.0}/flaml/automl/nlp/huggingface/__init__.py +0 -0
  85. {FLAML-2.1.1 → flaml-2.2.0}/flaml/automl/nlp/huggingface/trainer.py +0 -0
  86. {FLAML-2.1.1 → flaml-2.2.0}/flaml/automl/spark/configs.py +0 -0
  87. {FLAML-2.1.1 → flaml-2.2.0}/flaml/automl/task/__init__.py +0 -0
  88. {FLAML-2.1.1 → flaml-2.2.0}/flaml/automl/time_series/feature.py +1 -1
  89. {FLAML-2.1.1 → flaml-2.2.0}/flaml/automl/time_series/sklearn.py +1 -1
  90. {FLAML-2.1.1 → flaml-2.2.0}/flaml/automl/training_log.py +2 -2
  91. {FLAML-2.1.1 → flaml-2.2.0}/flaml/config.py +0 -0
  92. {FLAML-2.1.1 → flaml-2.2.0}/flaml/default/__init__.py +12 -12
  93. {FLAML-2.1.1 → flaml-2.2.0}/flaml/default/all/binary.json +0 -0
  94. {FLAML-2.1.1 → flaml-2.2.0}/flaml/default/all/multiclass.json +0 -0
  95. {FLAML-2.1.1 → flaml-2.2.0}/flaml/default/all/regression.json +0 -0
  96. {FLAML-2.1.1 → flaml-2.2.0}/flaml/default/extra_tree/binary.json +0 -0
  97. {FLAML-2.1.1 → flaml-2.2.0}/flaml/default/extra_tree/multiclass.json +0 -0
  98. {FLAML-2.1.1 → flaml-2.2.0}/flaml/default/extra_tree/regression.json +0 -0
  99. {FLAML-2.1.1 → flaml-2.2.0}/flaml/default/lgbm/binary.json +0 -0
  100. {FLAML-2.1.1 → flaml-2.2.0}/flaml/default/lgbm/multiclass.json +0 -0
  101. {FLAML-2.1.1 → flaml-2.2.0}/flaml/default/lgbm/regression.json +0 -0
  102. {FLAML-2.1.1 → flaml-2.2.0}/flaml/default/rf/binary.json +0 -0
  103. {FLAML-2.1.1 → flaml-2.2.0}/flaml/default/rf/multiclass.json +0 -0
  104. {FLAML-2.1.1 → flaml-2.2.0}/flaml/default/rf/regression.json +0 -0
  105. {FLAML-2.1.1 → flaml-2.2.0}/flaml/default/xgb_limitdepth/binary.json +0 -0
  106. {FLAML-2.1.1 → flaml-2.2.0}/flaml/default/xgb_limitdepth/multiclass.json +0 -0
  107. {FLAML-2.1.1 → flaml-2.2.0}/flaml/default/xgb_limitdepth/regression.json +0 -0
  108. {FLAML-2.1.1 → flaml-2.2.0}/flaml/default/xgboost/binary.json +0 -0
  109. {FLAML-2.1.1 → flaml-2.2.0}/flaml/default/xgboost/multiclass.json +0 -0
  110. {FLAML-2.1.1 → flaml-2.2.0}/flaml/default/xgboost/regression.json +0 -0
  111. {FLAML-2.1.1 → flaml-2.2.0}/flaml/onlineml/__init__.py +0 -0
  112. {FLAML-2.1.1 → flaml-2.2.0}/flaml/tune/result.py +0 -0
  113. {FLAML-2.1.1 → flaml-2.2.0}/flaml/tune/scheduler/__init__.py +2 -2
  114. {FLAML-2.1.1 → flaml-2.2.0}/flaml/tune/scheduler/trial_scheduler.py +0 -0
  115. {FLAML-2.1.1 → flaml-2.2.0}/flaml/tune/searcher/__init__.py +0 -0
  116. {FLAML-2.1.1 → flaml-2.2.0}/flaml/tune/spark/__init__.py +1 -1
  117. {FLAML-2.1.1 → flaml-2.2.0}/flaml/tune/trial_runner.py +1 -1
  118. {FLAML-2.1.1 → flaml-2.2.0}/flaml/tune/utils.py +0 -0
  119. {FLAML-2.1.1 → flaml-2.2.0}/setup.cfg +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: FLAML
3
- Version: 2.1.1
3
+ Version: 2.2.0
4
4
  Summary: A fast library for automated machine learning and tuning
5
5
  Home-page: https://github.com/microsoft/FLAML
6
6
  Author: Microsoft Corporation
@@ -10,29 +10,137 @@ Classifier: License :: OSI Approved :: MIT License
10
10
  Classifier: Operating System :: OS Independent
11
11
  Requires-Python: >=3.6
12
12
  Description-Content-Type: text/markdown
13
+ License-File: LICENSE
14
+ License-File: NOTICE.md
15
+ Requires-Dist: NumPy>=1.17
13
16
  Provides-Extra: automl
17
+ Requires-Dist: lightgbm>=2.3.1; extra == "automl"
18
+ Requires-Dist: xgboost<3.0.0,>=0.90; extra == "automl"
19
+ Requires-Dist: scipy>=1.4.1; extra == "automl"
20
+ Requires-Dist: pandas>=1.1.4; extra == "automl"
21
+ Requires-Dist: scikit-learn>=1.0.0; extra == "automl"
14
22
  Provides-Extra: notebook
23
+ Requires-Dist: jupyter; extra == "notebook"
15
24
  Provides-Extra: spark
25
+ Requires-Dist: pyspark>=3.2.0; extra == "spark"
26
+ Requires-Dist: joblibspark>=0.5.0; extra == "spark"
27
+ Requires-Dist: joblib<=1.3.2; extra == "spark"
16
28
  Provides-Extra: test
29
+ Requires-Dist: jupyter; extra == "test"
30
+ Requires-Dist: lightgbm>=2.3.1; extra == "test"
31
+ Requires-Dist: xgboost<2.0.0,>=0.90; extra == "test"
32
+ Requires-Dist: scipy>=1.4.1; extra == "test"
33
+ Requires-Dist: pandas>=1.1.4; extra == "test"
34
+ Requires-Dist: scikit-learn>=1.0.0; extra == "test"
35
+ Requires-Dist: thop; extra == "test"
36
+ Requires-Dist: pytest>=6.1.1; extra == "test"
37
+ Requires-Dist: coverage>=5.3; extra == "test"
38
+ Requires-Dist: pre-commit; extra == "test"
39
+ Requires-Dist: torch; extra == "test"
40
+ Requires-Dist: torchvision; extra == "test"
41
+ Requires-Dist: catboost<1.2,>=0.26; python_version < "3.11" and extra == "test"
42
+ Requires-Dist: catboost>=0.26; python_version >= "3.11" and extra == "test"
43
+ Requires-Dist: rgf-python; extra == "test"
44
+ Requires-Dist: optuna<=3.6.1,>=2.8.0; extra == "test"
45
+ Requires-Dist: openml; extra == "test"
46
+ Requires-Dist: statsmodels>=0.12.2; extra == "test"
47
+ Requires-Dist: psutil==5.8.0; extra == "test"
48
+ Requires-Dist: dataclasses; extra == "test"
49
+ Requires-Dist: transformers[torch]==4.26; extra == "test"
50
+ Requires-Dist: datasets; extra == "test"
51
+ Requires-Dist: nltk; extra == "test"
52
+ Requires-Dist: rouge_score; extra == "test"
53
+ Requires-Dist: hcrystalball==0.1.10; extra == "test"
54
+ Requires-Dist: seqeval; extra == "test"
55
+ Requires-Dist: pytorch-forecasting<=0.10.1,>=0.9.0; python_version < "3.11" and extra == "test"
56
+ Requires-Dist: mlflow; extra == "test"
57
+ Requires-Dist: pyspark>=3.2.0; extra == "test"
58
+ Requires-Dist: joblibspark>=0.5.0; extra == "test"
59
+ Requires-Dist: joblib<=1.3.2; extra == "test"
60
+ Requires-Dist: nbconvert; extra == "test"
61
+ Requires-Dist: nbformat; extra == "test"
62
+ Requires-Dist: ipykernel; extra == "test"
63
+ Requires-Dist: pytorch-lightning<1.9.1; extra == "test"
64
+ Requires-Dist: tensorboardX==2.6; extra == "test"
65
+ Requires-Dist: requests<2.29.0; extra == "test"
66
+ Requires-Dist: packaging; extra == "test"
67
+ Requires-Dist: pydantic==1.10.9; extra == "test"
68
+ Requires-Dist: sympy; extra == "test"
69
+ Requires-Dist: wolframalpha; extra == "test"
17
70
  Provides-Extra: catboost
71
+ Requires-Dist: catboost<1.2,>=0.26; python_version < "3.11" and extra == "catboost"
72
+ Requires-Dist: catboost<=1.2.5,>=0.26; python_version >= "3.11" and extra == "catboost"
18
73
  Provides-Extra: blendsearch
74
+ Requires-Dist: optuna<=3.6.1,>=2.8.0; extra == "blendsearch"
75
+ Requires-Dist: packaging; extra == "blendsearch"
19
76
  Provides-Extra: ray
77
+ Requires-Dist: ray[tune]~=1.13; extra == "ray"
20
78
  Provides-Extra: azureml
79
+ Requires-Dist: azureml-mlflow; extra == "azureml"
21
80
  Provides-Extra: nni
81
+ Requires-Dist: nni; extra == "nni"
22
82
  Provides-Extra: vw
83
+ Requires-Dist: vowpalwabbit<9.0.0,>=8.10.0; extra == "vw"
84
+ Requires-Dist: scikit-learn; extra == "vw"
23
85
  Provides-Extra: hf
86
+ Requires-Dist: transformers[torch]==4.26; extra == "hf"
87
+ Requires-Dist: datasets; extra == "hf"
88
+ Requires-Dist: nltk; extra == "hf"
89
+ Requires-Dist: rouge_score; extra == "hf"
90
+ Requires-Dist: seqeval; extra == "hf"
24
91
  Provides-Extra: nlp
25
- Provides-Extra: ts_forecast
92
+ Requires-Dist: transformers[torch]==4.26; extra == "nlp"
93
+ Requires-Dist: datasets; extra == "nlp"
94
+ Requires-Dist: nltk; extra == "nlp"
95
+ Requires-Dist: rouge_score; extra == "nlp"
96
+ Requires-Dist: seqeval; extra == "nlp"
97
+ Provides-Extra: ts-forecast
98
+ Requires-Dist: holidays<0.14; extra == "ts-forecast"
99
+ Requires-Dist: prophet>=1.0.1; extra == "ts-forecast"
100
+ Requires-Dist: statsmodels>=0.12.2; extra == "ts-forecast"
101
+ Requires-Dist: hcrystalball==0.1.10; extra == "ts-forecast"
26
102
  Provides-Extra: forecast
103
+ Requires-Dist: holidays<0.14; extra == "forecast"
104
+ Requires-Dist: prophet>=1.0.1; extra == "forecast"
105
+ Requires-Dist: statsmodels>=0.12.2; extra == "forecast"
106
+ Requires-Dist: hcrystalball==0.1.10; extra == "forecast"
107
+ Requires-Dist: pytorch-forecasting>=0.9.0; extra == "forecast"
108
+ Requires-Dist: pytorch-lightning==1.9.0; extra == "forecast"
109
+ Requires-Dist: tensorboardX==2.6; extra == "forecast"
27
110
  Provides-Extra: benchmark
111
+ Requires-Dist: catboost>=0.26; extra == "benchmark"
112
+ Requires-Dist: psutil==5.8.0; extra == "benchmark"
113
+ Requires-Dist: xgboost==1.3.3; extra == "benchmark"
114
+ Requires-Dist: pandas==1.1.4; extra == "benchmark"
28
115
  Provides-Extra: openai
116
+ Requires-Dist: openai==0.27.8; extra == "openai"
117
+ Requires-Dist: diskcache; extra == "openai"
29
118
  Provides-Extra: autogen
119
+ Requires-Dist: openai==0.27.8; extra == "autogen"
120
+ Requires-Dist: diskcache; extra == "autogen"
121
+ Requires-Dist: termcolor; extra == "autogen"
30
122
  Provides-Extra: mathchat
123
+ Requires-Dist: openai==0.27.8; extra == "mathchat"
124
+ Requires-Dist: diskcache; extra == "mathchat"
125
+ Requires-Dist: termcolor; extra == "mathchat"
126
+ Requires-Dist: sympy; extra == "mathchat"
127
+ Requires-Dist: pydantic==1.10.9; extra == "mathchat"
128
+ Requires-Dist: wolframalpha; extra == "mathchat"
31
129
  Provides-Extra: retrievechat
130
+ Requires-Dist: openai==0.27.8; extra == "retrievechat"
131
+ Requires-Dist: diskcache; extra == "retrievechat"
132
+ Requires-Dist: termcolor; extra == "retrievechat"
133
+ Requires-Dist: chromadb; extra == "retrievechat"
134
+ Requires-Dist: tiktoken; extra == "retrievechat"
135
+ Requires-Dist: sentence_transformers; extra == "retrievechat"
32
136
  Provides-Extra: synapse
137
+ Requires-Dist: joblibspark>=0.5.0; extra == "synapse"
138
+ Requires-Dist: optuna<=3.6.1,>=2.8.0; extra == "synapse"
139
+ Requires-Dist: pyspark>=3.2.0; extra == "synapse"
33
140
  Provides-Extra: autozero
34
- License-File: LICENSE
35
- License-File: NOTICE.md
141
+ Requires-Dist: scikit-learn; extra == "autozero"
142
+ Requires-Dist: pandas; extra == "autozero"
143
+ Requires-Dist: packaging; extra == "autozero"
36
144
 
37
145
  [![PyPI version](https://badge.fury.io/py/FLAML.svg)](https://badge.fury.io/py/FLAML)
38
146
  ![Conda version](https://img.shields.io/conda/vn/conda-forge/flaml)
@@ -40,8 +148,8 @@ License-File: NOTICE.md
40
148
  ![Python Version](https://img.shields.io/badge/3.8%20%7C%203.9%20%7C%203.10-blue)
41
149
  [![Downloads](https://pepy.tech/badge/flaml)](https://pepy.tech/project/flaml)
42
150
  [![](https://img.shields.io/discord/1025786666260111483?logo=discord&style=flat)](https://discord.gg/Cppx2vSPVP)
43
- <!-- [![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) -->
44
151
 
152
+ <!-- [![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) -->
45
153
 
46
154
  # A Fast Library for Automated Machine Learning & Tuning
47
155
 
@@ -50,7 +158,7 @@ License-File: NOTICE.md
50
158
  <br>
51
159
  </p>
52
160
 
53
- :fire: Heads-up: We're preparing to migrate [AutoGen](https://microsoft.github.io/autogen/) into a dedicated github repository. Alongside this move, we'll also launch a dedicated Discord server and a website for comprehensive documentation.
161
+ :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.
54
162
 
55
163
  :fire: The automated multi-agent chat framework in [AutoGen](https://microsoft.github.io/autogen/) is in preview from v2.0.0.
56
164
 
@@ -60,15 +168,15 @@ License-File: NOTICE.md
60
168
 
61
169
  :fire: FLAML supports Code-First AutoML & Tuning – Private Preview in [Microsoft Fabric Data Science](https://learn.microsoft.com/en-us/fabric/data-science/).
62
170
 
63
-
64
171
  ## What is FLAML
172
+
65
173
  FLAML is a lightweight Python library for efficient automation of machine
66
174
  learning and AI operations. It automates workflow based on large language models, machine learning models, etc.
67
175
  and optimizes their performance.
68
176
 
69
- * 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.
70
- * 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.
71
- * 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.
177
+ - 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.
178
+ - 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.
179
+ - 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.
72
180
 
73
181
  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.
74
182
 
@@ -83,6 +191,7 @@ pip install flaml
83
191
  ```
84
192
 
85
193
  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.
194
+
86
195
  ```bash
87
196
  pip install "flaml[autogen]"
88
197
  ```
@@ -92,18 +201,24 @@ Each of the [`notebook examples`](https://github.com/microsoft/FLAML/tree/main/n
92
201
 
93
202
  ## Quickstart
94
203
 
95
- * (New) The [autogen](https://microsoft.github.io/autogen/) package enables the next-gen GPT-X applications with a generic multi-agent conversation framework.
96
- It offers customizable and conversable agents which integrate LLMs, tools and human.
97
- 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,
204
+ - (New) The [autogen](https://microsoft.github.io/autogen/) package enables the next-gen GPT-X applications with a generic multi-agent conversation framework.
205
+ It offers customizable and conversable agents which integrate LLMs, tools and human.
206
+ 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,
207
+
98
208
  ```python
99
209
  from flaml import autogen
210
+
100
211
  assistant = autogen.AssistantAgent("assistant")
101
212
  user_proxy = autogen.UserProxyAgent("user_proxy")
102
- user_proxy.initiate_chat(assistant, message="Show me the YTD gain of 10 largest technology companies as of today.")
213
+ user_proxy.initiate_chat(
214
+ assistant,
215
+ message="Show me the YTD gain of 10 largest technology companies as of today.",
216
+ )
103
217
  # This initiates an automated chat between the two agents to solve the task
104
218
  ```
105
219
 
106
220
  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.
221
+
107
222
  ```python
108
223
  # perform tuning
109
224
  config, analysis = autogen.Completion.tune(
@@ -118,30 +233,32 @@ config, analysis = autogen.Completion.tune(
118
233
  # perform inference for a test instance
119
234
  response = autogen.Completion.create(context=test_instance, **config)
120
235
  ```
121
- * With three lines of code, you can start using this economical and fast
122
- AutoML engine as a [scikit-learn style estimator](https://microsoft.github.io/FLAML/docs/Use-Cases/Task-Oriented-AutoML).
236
+
237
+ - With three lines of code, you can start using this economical and fast
238
+ AutoML engine as a [scikit-learn style estimator](https://microsoft.github.io/FLAML/docs/Use-Cases/Task-Oriented-AutoML).
123
239
 
124
240
  ```python
125
241
  from flaml import AutoML
242
+
126
243
  automl = AutoML()
127
244
  automl.fit(X_train, y_train, task="classification")
128
245
  ```
129
246
 
130
- * You can restrict the learners and use FLAML as a fast hyperparameter tuning
131
- 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).
247
+ - You can restrict the learners and use FLAML as a fast hyperparameter tuning
248
+ 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).
132
249
 
133
250
  ```python
134
251
  automl.fit(X_train, y_train, task="classification", estimator_list=["lgbm"])
135
252
  ```
136
253
 
137
- * You can also run generic hyperparameter tuning for a [custom function](https://microsoft.github.io/FLAML/docs/Use-Cases/Tune-User-Defined-Function).
254
+ - You can also run generic hyperparameter tuning for a [custom function](https://microsoft.github.io/FLAML/docs/Use-Cases/Tune-User-Defined-Function).
138
255
 
139
256
  ```python
140
257
  from flaml import tune
141
258
  tune.run(evaluation_function, config={…}, low_cost_partial_config={…}, time_budget_s=3600)
142
259
  ```
143
260
 
144
- * [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.
261
+ - [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.
145
262
 
146
263
  ```python
147
264
  from flaml.default import LGBMRegressor
@@ -34,6 +34,8 @@ flaml/automl/ml.py
34
34
  flaml/automl/model.py
35
35
  flaml/automl/state.py
36
36
  flaml/automl/training_log.py
37
+ flaml/automl/contrib/__init__.py
38
+ flaml/automl/contrib/histgb.py
37
39
  flaml/automl/nlp/__init__.py
38
40
  flaml/automl/nlp/utils.py
39
41
  flaml/automl/nlp/huggingface/__init__.py
@@ -1,4 +1,4 @@
1
- NumPy>=1.17.0rc1
1
+ NumPy>=1.17
2
2
 
3
3
  [autogen]
4
4
  openai==0.27.8
@@ -7,10 +7,10 @@ termcolor
7
7
 
8
8
  [automl]
9
9
  lightgbm>=2.3.1
10
- xgboost>=0.90
10
+ xgboost<3.0.0,>=0.90
11
11
  scipy>=1.4.1
12
12
  pandas>=1.1.4
13
- scikit-learn>=0.24
13
+ scikit-learn>=1.0.0
14
14
 
15
15
  [autozero]
16
16
  scikit-learn
@@ -27,11 +27,16 @@ xgboost==1.3.3
27
27
  pandas==1.1.4
28
28
 
29
29
  [blendsearch]
30
- optuna==2.8.0
30
+ optuna<=3.6.1,>=2.8.0
31
31
  packaging
32
32
 
33
33
  [catboost]
34
- catboost>=0.26
34
+
35
+ [catboost:python_version < "3.11"]
36
+ catboost<1.2,>=0.26
37
+
38
+ [catboost:python_version >= "3.11"]
39
+ catboost<=1.2.5,>=0.26
35
40
 
36
41
  [forecast]
37
42
  holidays<0.14
@@ -88,29 +93,28 @@ sentence_transformers
88
93
  [spark]
89
94
  pyspark>=3.2.0
90
95
  joblibspark>=0.5.0
91
- joblib<1.3.0
96
+ joblib<=1.3.2
92
97
 
93
98
  [synapse]
94
99
  joblibspark>=0.5.0
95
- optuna==2.8.0
100
+ optuna<=3.6.1,>=2.8.0
96
101
  pyspark>=3.2.0
97
- joblib<1.3.0
98
102
 
99
103
  [test]
104
+ jupyter
100
105
  lightgbm>=2.3.1
101
- xgboost>=0.90
106
+ xgboost<2.0.0,>=0.90
102
107
  scipy>=1.4.1
103
108
  pandas>=1.1.4
104
- scikit-learn>=0.24
109
+ scikit-learn>=1.0.0
105
110
  thop
106
111
  pytest>=6.1.1
107
112
  coverage>=5.3
108
113
  pre-commit
109
114
  torch
110
115
  torchvision
111
- catboost<1.2,>=0.26
112
116
  rgf-python
113
- optuna==2.8.0
117
+ optuna<=3.6.1,>=2.8.0
114
118
  openml
115
119
  statsmodels>=0.12.2
116
120
  psutil==5.8.0
@@ -121,10 +125,10 @@ nltk
121
125
  rouge_score
122
126
  hcrystalball==0.1.10
123
127
  seqeval
124
- pytorch-forecasting<=0.10.1,>=0.9.0
125
128
  mlflow
126
129
  pyspark>=3.2.0
127
130
  joblibspark>=0.5.0
131
+ joblib<=1.3.2
128
132
  nbconvert
129
133
  nbformat
130
134
  ipykernel
@@ -135,7 +139,13 @@ packaging
135
139
  pydantic==1.10.9
136
140
  sympy
137
141
  wolframalpha
138
- joblib<1.3.0
142
+
143
+ [test:python_version < "3.11"]
144
+ catboost<1.2,>=0.26
145
+ pytorch-forecasting<=0.10.1,>=0.9.0
146
+
147
+ [test:python_version >= "3.11"]
148
+ catboost>=0.26
139
149
 
140
150
  [ts_forecast]
141
151
  holidays<0.14
flaml-2.2.0/NOTICE.md ADDED
@@ -0,0 +1,295 @@
1
+ # NOTICES
2
+
3
+ This repository incorporates material as listed below or described in the code.
4
+
5
+ ## Component. Ray.
6
+
7
+ Code in tune/\[analysis.py, sample.py, trial.py, result.py\],
8
+ searcher/\[suggestion.py, variant_generator.py\], and scheduler/trial_scheduler.py is adapted from
9
+ https://github.com/ray-project/ray/blob/master/python/ray/tune/
10
+
11
+ ## Open Source License/Copyright Notice.
12
+
13
+ Apache License
14
+ Version 2.0, January 2004
15
+ http://www.apache.org/licenses/
16
+
17
+ TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
18
+
19
+ 1. Definitions.
20
+
21
+ "License" shall mean the terms and conditions for use, reproduction,
22
+ and distribution as defined by Sections 1 through 9 of this document.
23
+
24
+ "Licensor" shall mean the copyright owner or entity authorized by
25
+ the copyright owner that is granting the License.
26
+
27
+ "Legal Entity" shall mean the union of the acting entity and all
28
+ other entities that control, are controlled by, or are under common
29
+ control with that entity. For the purposes of this definition,
30
+ "control" means (i) the power, direct or indirect, to cause the
31
+ direction or management of such entity, whether by contract or
32
+ otherwise, or (ii) ownership of fifty percent (50%) or more of the
33
+ outstanding shares, or (iii) beneficial ownership of such entity.
34
+
35
+ "You" (or "Your") shall mean an individual or Legal Entity
36
+ exercising permissions granted by this License.
37
+
38
+ "Source" form shall mean the preferred form for making modifications,
39
+ including but not limited to software source code, documentation
40
+ source, and configuration files.
41
+
42
+ "Object" form shall mean any form resulting from mechanical
43
+ transformation or translation of a Source form, including but
44
+ not limited to compiled object code, generated documentation,
45
+ and conversions to other media types.
46
+
47
+ "Work" shall mean the work of authorship, whether in Source or
48
+ Object form, made available under the License, as indicated by a
49
+ copyright notice that is included in or attached to the work
50
+ (an example is provided in the Appendix below).
51
+
52
+ "Derivative Works" shall mean any work, whether in Source or Object
53
+ form, that is based on (or derived from) the Work and for which the
54
+ editorial revisions, annotations, elaborations, or other modifications
55
+ represent, as a whole, an original work of authorship. For the purposes
56
+ of this License, Derivative Works shall not include works that remain
57
+ separable from, or merely link (or bind by name) to the interfaces of,
58
+ the Work and Derivative Works thereof.
59
+
60
+ "Contribution" shall mean any work of authorship, including
61
+ the original version of the Work and any modifications or additions
62
+ to that Work or Derivative Works thereof, that is intentionally
63
+ submitted to Licensor for inclusion in the Work by the copyright owner
64
+ or by an individual or Legal Entity authorized to submit on behalf of
65
+ the copyright owner. For the purposes of this definition, "submitted"
66
+ means any form of electronic, verbal, or written communication sent
67
+ to the Licensor or its representatives, including but not limited to
68
+ communication on electronic mailing lists, source code control systems,
69
+ and issue tracking systems that are managed by, or on behalf of, the
70
+ Licensor for the purpose of discussing and improving the Work, but
71
+ excluding communication that is conspicuously marked or otherwise
72
+ designated in writing by the copyright owner as "Not a Contribution."
73
+
74
+ "Contributor" shall mean Licensor and any individual or Legal Entity
75
+ on behalf of whom a Contribution has been received by Licensor and
76
+ subsequently incorporated within the Work.
77
+
78
+ 1. Grant of Copyright License. Subject to the terms and conditions of
79
+ this License, each Contributor hereby grants to You a perpetual,
80
+ worldwide, non-exclusive, no-charge, royalty-free, irrevocable
81
+ copyright license to reproduce, prepare Derivative Works of,
82
+ publicly display, publicly perform, sublicense, and distribute the
83
+ Work and such Derivative Works in Source or Object form.
84
+
85
+ 1. Grant of Patent License. Subject to the terms and conditions of
86
+ this License, each Contributor hereby grants to You a perpetual,
87
+ worldwide, non-exclusive, no-charge, royalty-free, irrevocable
88
+ (except as stated in this section) patent license to make, have made,
89
+ use, offer to sell, sell, import, and otherwise transfer the Work,
90
+ where such license applies only to those patent claims licensable
91
+ by such Contributor that are necessarily infringed by their
92
+ Contribution(s) alone or by combination of their Contribution(s)
93
+ with the Work to which such Contribution(s) was submitted. If You
94
+ institute patent litigation against any entity (including a
95
+ cross-claim or counterclaim in a lawsuit) alleging that the Work
96
+ or a Contribution incorporated within the Work constitutes direct
97
+ or contributory patent infringement, then any patent licenses
98
+ granted to You under this License for that Work shall terminate
99
+ as of the date such litigation is filed.
100
+
101
+ 1. Redistribution. You may reproduce and distribute copies of the
102
+ Work or Derivative Works thereof in any medium, with or without
103
+ modifications, and in Source or Object form, provided that You
104
+ meet the following conditions:
105
+
106
+ (a) You must give any other recipients of the Work or
107
+ Derivative Works a copy of this License; and
108
+
109
+ (b) You must cause any modified files to carry prominent notices
110
+ stating that You changed the files; and
111
+
112
+ (c) You must retain, in the Source form of any Derivative Works
113
+ that You distribute, all copyright, patent, trademark, and
114
+ attribution notices from the Source form of the Work,
115
+ excluding those notices that do not pertain to any part of
116
+ the Derivative Works; and
117
+
118
+ (d) If the Work includes a "NOTICE" text file as part of its
119
+ distribution, then any Derivative Works that You distribute must
120
+ include a readable copy of the attribution notices contained
121
+ within such NOTICE file, excluding those notices that do not
122
+ pertain to any part of the Derivative Works, in at least one
123
+ of the following places: within a NOTICE text file distributed
124
+ as part of the Derivative Works; within the Source form or
125
+ documentation, if provided along with the Derivative Works; or,
126
+ within a display generated by the Derivative Works, if and
127
+ wherever such third-party notices normally appear. The contents
128
+ of the NOTICE file are for informational purposes only and
129
+ do not modify the License. You may add Your own attribution
130
+ notices within Derivative Works that You distribute, alongside
131
+ or as an addendum to the NOTICE text from the Work, provided
132
+ that such additional attribution notices cannot be construed
133
+ as modifying the License.
134
+
135
+ You may add Your own copyright statement to Your modifications and
136
+ may provide additional or different license terms and conditions
137
+ for use, reproduction, or distribution of Your modifications, or
138
+ for any such Derivative Works as a whole, provided Your use,
139
+ reproduction, and distribution of the Work otherwise complies with
140
+ the conditions stated in this License.
141
+
142
+ 1. Submission of Contributions. Unless You explicitly state otherwise,
143
+ any Contribution intentionally submitted for inclusion in the Work
144
+ by You to the Licensor shall be under the terms and conditions of
145
+ this License, without any additional terms or conditions.
146
+ Notwithstanding the above, nothing herein shall supersede or modify
147
+ the terms of any separate license agreement you may have executed
148
+ with Licensor regarding such Contributions.
149
+
150
+ 1. Trademarks. This License does not grant permission to use the trade
151
+ names, trademarks, service marks, or product names of the Licensor,
152
+ except as required for reasonable and customary use in describing the
153
+ origin of the Work and reproducing the content of the NOTICE file.
154
+
155
+ 1. Disclaimer of Warranty. Unless required by applicable law or
156
+ agreed to in writing, Licensor provides the Work (and each
157
+ Contributor provides its Contributions) on an "AS IS" BASIS,
158
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
159
+ implied, including, without limitation, any warranties or conditions
160
+ of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
161
+ PARTICULAR PURPOSE. You are solely responsible for determining the
162
+ appropriateness of using or redistributing the Work and assume any
163
+ risks associated with Your exercise of permissions under this License.
164
+
165
+ 1. Limitation of Liability. In no event and under no legal theory,
166
+ whether in tort (including negligence), contract, or otherwise,
167
+ unless required by applicable law (such as deliberate and grossly
168
+ negligent acts) or agreed to in writing, shall any Contributor be
169
+ liable to You for damages, including any direct, indirect, special,
170
+ incidental, or consequential damages of any character arising as a
171
+ result of this License or out of the use or inability to use the
172
+ Work (including but not limited to damages for loss of goodwill,
173
+ work stoppage, computer failure or malfunction, or any and all
174
+ other commercial damages or losses), even if such Contributor
175
+ has been advised of the possibility of such damages.
176
+
177
+ 1. Accepting Warranty or Additional Liability. While redistributing
178
+ the Work or Derivative Works thereof, You may choose to offer,
179
+ and charge a fee for, acceptance of support, warranty, indemnity,
180
+ or other liability obligations and/or rights consistent with this
181
+ License. However, in accepting such obligations, You may act only
182
+ on Your own behalf and on Your sole responsibility, not on behalf
183
+ of any other Contributor, and only if You agree to indemnify,
184
+ defend, and hold each Contributor harmless for any liability
185
+ incurred by, or claims asserted against, such Contributor by reason
186
+ of your accepting any such warranty or additional liability.
187
+
188
+ END OF TERMS AND CONDITIONS
189
+
190
+ APPENDIX: How to apply the Apache License to your work.
191
+
192
+ ```
193
+ To apply the Apache License to your work, attach the following
194
+ boilerplate notice, with the fields enclosed by brackets "{}"
195
+ replaced with your own identifying information. (Don't include
196
+ the brackets!) The text should be enclosed in the appropriate
197
+ comment syntax for the file format. We also recommend that a
198
+ file or class name and description of purpose be included on the
199
+ same "printed page" as the copyright notice for easier
200
+ identification within third-party archives.
201
+ ```
202
+
203
+ Copyright {yyyy} {name of copyright owner}
204
+
205
+ Licensed under the Apache License, Version 2.0 (the "License");
206
+ you may not use this file except in compliance with the License.
207
+ You may obtain a copy of the License at
208
+
209
+ ```
210
+ http://www.apache.org/licenses/LICENSE-2.0
211
+ ```
212
+
213
+ Unless required by applicable law or agreed to in writing, software
214
+ distributed under the License is distributed on an "AS IS" BASIS,
215
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
216
+ See the License for the specific language governing permissions and
217
+ limitations under the License.
218
+
219
+ ______________________________________________________________________
220
+
221
+ Code in python/ray/rllib/{evolution_strategies, dqn} adapted from
222
+ https://github.com/openai (MIT License)
223
+
224
+ Copyright (c) 2016 OpenAI (http://openai.com)
225
+
226
+ Permission is hereby granted, free of charge, to any person obtaining a copy
227
+ of this software and associated documentation files (the "Software"), to deal
228
+ in the Software without restriction, including without limitation the rights
229
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
230
+ copies of the Software, and to permit persons to whom the Software is
231
+ furnished to do so, subject to the following conditions:
232
+
233
+ The above copyright notice and this permission notice shall be included in
234
+ all copies or substantial portions of the Software.
235
+
236
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
237
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
238
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
239
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
240
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
241
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
242
+ THE SOFTWARE.
243
+
244
+ ______________________________________________________________________
245
+
246
+ Code in python/ray/rllib/impala/vtrace.py from
247
+ https://github.com/deepmind/scalable_agent
248
+
249
+ Copyright 2018 Google LLC
250
+
251
+ Licensed under the Apache License, Version 2.0 (the "License");
252
+ you may not use this file except in compliance with the License.
253
+ You may obtain a copy of the License at
254
+
255
+ ```
256
+ https://www.apache.org/licenses/LICENSE-2.0
257
+ ```
258
+
259
+ Unless required by applicable law or agreed to in writing, software
260
+ distributed under the License is distributed on an "AS IS" BASIS,
261
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
262
+ See the License for the specific language governing permissions and
263
+ limitations under the License.
264
+
265
+ ______________________________________________________________________
266
+
267
+ Code in python/ray/rllib/ars is adapted from https://github.com/modestyachts/ARS
268
+
269
+ Copyright (c) 2018, ARS contributors (Horia Mania, Aurelia Guy, Benjamin Recht)
270
+ All rights reserved.
271
+
272
+ Redistribution and use of ARS in source and binary forms, with or without
273
+ modification, are permitted provided that the following conditions are met:
274
+
275
+ 1. Redistributions of source code must retain the above copyright notice, this
276
+ list of conditions and the following disclaimer.
277
+
278
+ 1. Redistributions in binary form must reproduce the above copyright notice,
279
+ this list of conditions and the following disclaimer in the documentation and/or
280
+ other materials provided with the distribution.
281
+
282
+ THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
283
+ ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
284
+ WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
285
+ DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR
286
+ ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
287
+ (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
288
+ LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
289
+ ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
290
+ (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
291
+ SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
292
+
293
+ ______________________________________________________________________
294
+
295
+ Code in python/ray/\_private/prometheus_exporter.py is adapted from https://github.com/census-instrumentation/opencensus-python/blob/master/contrib/opencensus-ext-prometheus/opencensus/ext/prometheus/stats_exporter/__init__.py