autogluon.core 1.2.1b20250416__tar.gz → 1.2.1b20250417__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 (100) hide show
  1. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/PKG-INFO +1 -1
  2. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/hpo/executors.py +6 -3
  3. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/metrics/classification_metrics.py +8 -1
  4. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/version.py +1 -1
  5. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon.core.egg-info/PKG-INFO +1 -1
  6. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon.core.egg-info/requires.txt +5 -5
  7. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/setup.cfg +0 -0
  8. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/setup.py +0 -0
  9. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/__init__.py +0 -0
  10. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/_setup_utils.py +0 -0
  11. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/augmentation/__init__.py +0 -0
  12. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/augmentation/distill_utils.py +0 -0
  13. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/calibrate/__init__.py +0 -0
  14. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/calibrate/_decision_threshold.py +0 -0
  15. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/calibrate/conformity_score.py +0 -0
  16. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/calibrate/temperature_scaling.py +0 -0
  17. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/callbacks/__init__.py +0 -0
  18. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/callbacks/_abstract_callback.py +0 -0
  19. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/callbacks/_early_stopping_callback.py +0 -0
  20. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/callbacks/_early_stopping_ensemble_callback.py +0 -0
  21. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/callbacks/_example_callback.py +0 -0
  22. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/constants.py +0 -0
  23. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/data/__init__.py +0 -0
  24. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/data/cleaner.py +0 -0
  25. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/data/label_cleaner.py +0 -0
  26. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/hpo/__init__.py +0 -0
  27. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/hpo/constants.py +0 -0
  28. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/hpo/exceptions.py +0 -0
  29. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/hpo/ray_hpo.py +0 -0
  30. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/hpo/ray_tune_constants.py +0 -0
  31. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/hpo/ray_tune_scheduler.py +0 -0
  32. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/hpo/ray_tune_scheduler_factory.py +0 -0
  33. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/hpo/ray_tune_searcher_factory.py +0 -0
  34. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/hpo/space_converter.py +0 -0
  35. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/learner/__init__.py +0 -0
  36. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/learner/abstract_learner.py +0 -0
  37. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/learning_curves/plot_curves.py +0 -0
  38. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/metrics/__init__.py +0 -0
  39. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/metrics/quantile_metrics.py +0 -0
  40. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/metrics/score_func.py +0 -0
  41. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/metrics/softclass_metrics.py +0 -0
  42. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/models/__init__.py +0 -0
  43. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/models/_utils.py +0 -0
  44. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/models/abstract/__init__.py +0 -0
  45. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/models/abstract/_tags.py +0 -0
  46. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/models/abstract/abstract_model.py +0 -0
  47. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/models/abstract/abstract_nn_model.py +0 -0
  48. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/models/abstract/model_trial.py +0 -0
  49. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/models/dummy/__init__.py +0 -0
  50. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/models/dummy/_dummy_quantile_regressor.py +0 -0
  51. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/models/dummy/dummy_model.py +0 -0
  52. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/models/ensemble/__init__.py +0 -0
  53. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/models/ensemble/bagged_ensemble_model.py +0 -0
  54. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/models/ensemble/fold_fitting_strategy.py +0 -0
  55. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/models/ensemble/ray_parallel_fold_fitting_strategy.py +0 -0
  56. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/models/ensemble/stacker_ensemble_model.py +0 -0
  57. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/models/ensemble/weighted_ensemble_model.py +0 -0
  58. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/models/greedy_ensemble/__init__.py +0 -0
  59. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/models/greedy_ensemble/ensemble_selection.py +0 -0
  60. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/models/greedy_ensemble/greedy_weighted_ensemble_model.py +0 -0
  61. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/problem_type.py +0 -0
  62. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/pseudolabeling/__init__.py +0 -0
  63. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/pseudolabeling/pseudolabeling.py +0 -0
  64. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/ray/__init__.py +0 -0
  65. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/ray/distributed_jobs_managers.py +0 -0
  66. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/ray/resources_calculator.py +0 -0
  67. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/scheduler/__init__.py +0 -0
  68. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/scheduler/reporter.py +0 -0
  69. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/scheduler/scheduler_factory.py +0 -0
  70. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/scheduler/seq_scheduler.py +0 -0
  71. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/searcher/__init__.py +0 -0
  72. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/searcher/dummy_searcher.py +0 -0
  73. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/searcher/exceptions.py +0 -0
  74. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/searcher/local_grid_searcher.py +0 -0
  75. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/searcher/local_random_searcher.py +0 -0
  76. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/searcher/local_searcher.py +0 -0
  77. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/searcher/searcher_factory.py +0 -0
  78. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/stacked_overfitting/utils.py +0 -0
  79. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/trainer/__init__.py +0 -0
  80. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/trainer/abstract_trainer.py +0 -0
  81. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/trainer/utils.py +0 -0
  82. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/utils/__init__.py +0 -0
  83. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/utils/decorators.py +0 -0
  84. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/utils/early_stopping.py +0 -0
  85. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/utils/exceptions.py +0 -0
  86. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/utils/feature_selection.py +0 -0
  87. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/utils/files.py +0 -0
  88. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/utils/infer_utils.py +0 -0
  89. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/utils/loaders/__init__.py +0 -0
  90. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/utils/miscs.py +0 -0
  91. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/utils/plots.py +0 -0
  92. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/utils/savers/__init__.py +0 -0
  93. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/utils/time.py +0 -0
  94. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/utils/utils.py +0 -0
  95. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon/core/utils/version_utils.py +0 -0
  96. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon.core.egg-info/SOURCES.txt +0 -0
  97. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon.core.egg-info/dependency_links.txt +0 -0
  98. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon.core.egg-info/namespace_packages.txt +0 -0
  99. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon.core.egg-info/top_level.txt +0 -0
  100. {autogluon.core-1.2.1b20250416 → autogluon.core-1.2.1b20250417}/src/autogluon.core.egg-info/zip-safe +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: autogluon.core
3
- Version: 1.2.1b20250416
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
@@ -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)
@@ -1,4 +1,4 @@
1
1
  """This is the autogluon version file."""
2
2
 
3
- __version__ = "1.2.1b20250416"
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.1b20250416
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
@@ -7,13 +7,13 @@ tqdm<5,>=4.38
7
7
  requests
8
8
  matplotlib<3.11,>=3.7.0
9
9
  boto3<2,>=1.10
10
- autogluon.common==1.2.1b20250416
10
+ autogluon.common==1.2.1b20250417
11
11
 
12
12
  [all]
13
- pyarrow>=15.0.0
14
13
  hyperopt<0.2.8,>=0.2.7
15
14
  ray[default,tune]<2.45,>=2.10.0
16
15
  ray[default]<2.45,>=2.10.0
16
+ pyarrow>=15.0.0
17
17
 
18
18
  [ray]
19
19
  ray[default]<2.45,>=2.10.0
@@ -24,8 +24,8 @@ ray[default,tune]<2.45,>=2.10.0
24
24
  hyperopt<0.2.8,>=0.2.7
25
25
 
26
26
  [tests]
27
- types-requests
28
- pytest-mypy
29
27
  pytest
30
- flake8
31
28
  types-setuptools
29
+ types-requests
30
+ flake8
31
+ pytest-mypy