autogluon.core 1.2.1b20250415__py3-none-any.whl → 1.2.1b20250417__py3-none-any.whl

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.
@@ -6,7 +6,7 @@ import math
6
6
  import os
7
7
  import time
8
8
  from abc import ABC, abstractmethod
9
- from typing import TYPE_CHECKING, Any, Callable, Dict, List, Optional, Tuple, Union
9
+ from typing import TYPE_CHECKING, Any, Callable, Dict, Optional, Tuple, Union
10
10
 
11
11
  import pandas as pd
12
12
 
@@ -14,7 +14,6 @@ from autogluon.common import space
14
14
  from autogluon.common.utils.resource_utils import ResourceManager
15
15
  from autogluon.common.utils.s3_utils import is_s3_url
16
16
 
17
- from ..ray.resources_calculator import ResourceCalculator
18
17
  from ..scheduler.scheduler_factory import scheduler_factory
19
18
  from ..utils.savers import save_pkl
20
19
  from .constants import CUSTOM_BACKEND, RAY_BACKEND
@@ -135,7 +134,11 @@ class HpoExecutor(ABC):
135
134
  resources_per_trial = self.hyperparameter_tune_kwargs["resources_per_trial"]
136
135
  total_num_cpus_per_trial = resources_per_trial.get("num_cpus")
137
136
  total_num_gpus_per_trial = resources_per_trial.get("num_gpus")
138
- user_specified_fold_resources = model_base._user_params_aux
137
+
138
+ if hasattr(model_base, "_user_params_aux"):
139
+ user_specified_fold_resources = model_base._user_params_aux
140
+ else:
141
+ user_specified_fold_resources = {}
139
142
  user_specified_fold_num_cpus = user_specified_fold_resources.get("num_cpus", None) # We shouldn't always use it
140
143
  user_specified_fold_num_gpus = user_specified_fold_resources.get("num_gpus", None)
141
144
  if user_specified_fold_num_cpus is not None or user_specified_fold_num_gpus is not None:
@@ -14,6 +14,13 @@ try:
14
14
  except:
15
15
  from sklearn.metrics.classification import _check_targets, type_of_target
16
16
 
17
+ try:
18
+ # numpy>=2
19
+ from numpy import trapezoid as trapezoid
20
+ except:
21
+ # numpy<2, deprecated in numpy>=2
22
+ from numpy import trapz as trapezoid
23
+
17
24
  logger = logging.getLogger(__name__)
18
25
 
19
26
 
@@ -434,4 +441,4 @@ def customized_binary_roc_auc_score(y_true: Union[np.array, pd.Series], y_score:
434
441
  raise ValueError("Only one class present in y_true. ROC AUC score is not defined in that case.")
435
442
  fpr = fps / fps[-1]
436
443
  tpr = tps / tps[-1]
437
- return np.trapz(tpr, fpr)
444
+ return trapezoid(tpr, fpr)
autogluon/core/version.py CHANGED
@@ -1,4 +1,4 @@
1
1
  """This is the autogluon version file."""
2
2
 
3
- __version__ = "1.2.1b20250415"
3
+ __version__ = "1.2.1b20250417"
4
4
  __lite__ = False
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: autogluon.core
3
- Version: 1.2.1b20250415
3
+ Version: 1.2.1b20250417
4
4
  Summary: Fast and Accurate ML in 3 Lines of Code
5
5
  Home-page: https://github.com/autogluon/autogluon
6
6
  Author: AutoGluon Community
@@ -45,12 +45,12 @@ Requires-Dist: tqdm<5,>=4.38
45
45
  Requires-Dist: requests
46
46
  Requires-Dist: matplotlib<3.11,>=3.7.0
47
47
  Requires-Dist: boto3<2,>=1.10
48
- Requires-Dist: autogluon.common==1.2.1b20250415
48
+ Requires-Dist: autogluon.common==1.2.1b20250417
49
49
  Provides-Extra: all
50
- Requires-Dist: ray[default]<2.45,>=2.10.0; extra == "all"
50
+ Requires-Dist: hyperopt<0.2.8,>=0.2.7; extra == "all"
51
51
  Requires-Dist: ray[default,tune]<2.45,>=2.10.0; extra == "all"
52
+ Requires-Dist: ray[default]<2.45,>=2.10.0; extra == "all"
52
53
  Requires-Dist: pyarrow>=15.0.0; extra == "all"
53
- Requires-Dist: hyperopt<0.2.8,>=0.2.7; extra == "all"
54
54
  Provides-Extra: ray
55
55
  Requires-Dist: ray[default]<2.45,>=2.10.0; extra == "ray"
56
56
  Provides-Extra: raytune
@@ -58,11 +58,11 @@ Requires-Dist: pyarrow>=15.0.0; extra == "raytune"
58
58
  Requires-Dist: ray[default,tune]<2.45,>=2.10.0; extra == "raytune"
59
59
  Requires-Dist: hyperopt<0.2.8,>=0.2.7; extra == "raytune"
60
60
  Provides-Extra: tests
61
- Requires-Dist: flake8; extra == "tests"
62
- Requires-Dist: pytest-mypy; extra == "tests"
63
- Requires-Dist: types-setuptools; extra == "tests"
64
61
  Requires-Dist: pytest; extra == "tests"
62
+ Requires-Dist: types-setuptools; extra == "tests"
65
63
  Requires-Dist: types-requests; extra == "tests"
64
+ Requires-Dist: flake8; extra == "tests"
65
+ Requires-Dist: pytest-mypy; extra == "tests"
66
66
 
67
67
 
68
68
 
@@ -1,9 +1,9 @@
1
- autogluon.core-1.2.1b20250415-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
1
+ autogluon.core-1.2.1b20250417-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
2
2
  autogluon/core/__init__.py,sha256=8KfvvHzXX3a4q6z43Dw1yE7VtbAoiSMaglVpKDy6Xeg,245
3
3
  autogluon/core/_setup_utils.py,sha256=TnDboM66MzSp0CLACo_vcNNwhORQWE8JL0kt1IYa7dI,7014
4
4
  autogluon/core/constants.py,sha256=nEVLdSFJ-5O-tz3jUD3qPX65RMp7g8qOR38XlurbP4Y,3403
5
5
  autogluon/core/problem_type.py,sha256=XJmMgeNBgS7u43pDK-spTivatPyh_INOXveEXwQt-Rw,2993
6
- autogluon/core/version.py,sha256=b5iA_UuQQbMm77xD8yblOR_jEE8jGtdLXDEMT2oS0Lk,91
6
+ autogluon/core/version.py,sha256=9DNociU3CkjmMks3rFL-2UfsFfw26T7_cq6VAom-ic4,91
7
7
  autogluon/core/augmentation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
8
8
  autogluon/core/augmentation/distill_utils.py,sha256=JBlp2WOMNKoJv8aKVwJVRQSalSk8jx36HM7-k_VvkhY,9404
9
9
  autogluon/core/calibrate/__init__.py,sha256=eU6qLj7DKUhaz2HHNHDrfroRaLM-mhuSncK_v1UP4F8,62
@@ -21,7 +21,7 @@ autogluon/core/data/label_cleaner.py,sha256=rgXFICs3oNlkaglnSohwjh-vcSaN2mcJTDAJ
21
21
  autogluon/core/hpo/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
22
22
  autogluon/core/hpo/constants.py,sha256=vYeJNySZtldqKX3hctMM-RMT8UuVZneH3tAneiIaIZM,117
23
23
  autogluon/core/hpo/exceptions.py,sha256=TvkCQZSd-TMwIqmxkOAONMuiz-HimxwHwjvGynOj5ug,44
24
- autogluon/core/hpo/executors.py,sha256=LUC5ALhJOhOdbMZLNv4Xik4jC-453tGF9yvgYU76apg,30269
24
+ autogluon/core/hpo/executors.py,sha256=uXt-xR9imDBx9Uf4E03QcLZqYfzrnyPPvj5yuhLQuYg,30335
25
25
  autogluon/core/hpo/ray_hpo.py,sha256=6Dn5SRo7xeeB1B-B7bryOAgkO7yh_Z2QLDx5enNkk9U,21523
26
26
  autogluon/core/hpo/ray_tune_constants.py,sha256=Lfx12RIRhSl9WgVpCQqEEtwWtvBIr6SCXNmeYuKZ88k,220
27
27
  autogluon/core/hpo/ray_tune_scheduler.py,sha256=T9xwB3FjTTnAEOhiqkB4eev21lQKU6K4hsx_WsS9-Lg,1112
@@ -32,7 +32,7 @@ autogluon/core/learner/__init__.py,sha256=pAEQBcY-xq-HG2QbHdt9V4iq1RSlz6gfcxNa40
32
32
  autogluon/core/learner/abstract_learner.py,sha256=A4nAR-U27wInLtghQjpLdEhBgh9M_2DFz7NhoVhjuL4,5784
33
33
  autogluon/core/learning_curves/plot_curves.py,sha256=GxWeX6gvp9XlA2QwtBi9cdaxjtbY3mSngWXl-EjM8bE,3141
34
34
  autogluon/core/metrics/__init__.py,sha256=7UTGdSwuRG4Wjfenyvj9RBYMtTSGAyPpy0Q_ktF-AF8,29827
35
- autogluon/core/metrics/classification_metrics.py,sha256=kKbq1cBMmc_gEm3AfXBCTk5KJDcrzcg-6gmQ_wONhFU,18342
35
+ autogluon/core/metrics/classification_metrics.py,sha256=-1osD_IH7fOA7O-iDAJwYpY0a4ZeD451M82GnwXSxJA,18496
36
36
  autogluon/core/metrics/quantile_metrics.py,sha256=yjnKSMuLdee4vemXIoxitCJWf_3lvE-RXlJ4V1371w4,1393
37
37
  autogluon/core/metrics/score_func.py,sha256=RIFOZMhPKYRFXBufoe5XIY1BRDr8aqGA-2zVwuTKi0A,4684
38
38
  autogluon/core/metrics/softclass_metrics.py,sha256=inn35DfftLZey0mK3OuMJPzm58eZgQ38vnEBO8sZ3mg,1272
@@ -89,11 +89,11 @@ autogluon/core/utils/utils.py,sha256=FMa9kIUAxA3IIBbATmBnNEVObSAivehZ2_zCy3PRR-c
89
89
  autogluon/core/utils/version_utils.py,sha256=5-r8hLRKTaZbj5qo2uzE_2E4casH49Ye3WyeHlgHuz4,3252
90
90
  autogluon/core/utils/loaders/__init__.py,sha256=W5FAdQvpDcn_uisqJrlSAObWVta-YjJLKGN3NCbEgIo,109
91
91
  autogluon/core/utils/savers/__init__.py,sha256=bGWciSxAkj6u06vOC4pTvr22f_1ey0glgvmjCMEOm78,89
92
- autogluon.core-1.2.1b20250415.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
93
- autogluon.core-1.2.1b20250415.dist-info/METADATA,sha256=Fis-Y0pbyB-j4OQUqdJf21V0Zwaar2PEvFZLPbjRf78,12399
94
- autogluon.core-1.2.1b20250415.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
95
- autogluon.core-1.2.1b20250415.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
96
- autogluon.core-1.2.1b20250415.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
97
- autogluon.core-1.2.1b20250415.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
98
- autogluon.core-1.2.1b20250415.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
99
- autogluon.core-1.2.1b20250415.dist-info/RECORD,,
92
+ autogluon.core-1.2.1b20250417.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
93
+ autogluon.core-1.2.1b20250417.dist-info/METADATA,sha256=WFLhQI8Kh5SJ8AdlKK6wvNcZrFZlMqmTjsJ8gtqGe6s,12399
94
+ autogluon.core-1.2.1b20250417.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
95
+ autogluon.core-1.2.1b20250417.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
96
+ autogluon.core-1.2.1b20250417.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
97
+ autogluon.core-1.2.1b20250417.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
98
+ autogluon.core-1.2.1b20250417.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
99
+ autogluon.core-1.2.1b20250417.dist-info/RECORD,,