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