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

autogluon/core/models/abstract/abstract_model.py

@@ -868,6 +868,15 @@ class AbstractModel(ModelBase, Tunable):
         minimum_model_num_cpus = minimum_model_resources.get("num_cpus", 1)
         minimum_model_num_gpus = minimum_model_resources.get("num_gpus", 0)
 
+        maximum_model_resources = self._get_maximum_resources()
+        maximum_model_num_cpus = maximum_model_resources.get("num_cpus", None)
+        maximum_model_num_gpus = maximum_model_resources.get("num_gpus", None)
+
+        if maximum_model_num_cpus is not None and maximum_model_num_cpus < num_cpus:
+            num_cpus = maximum_model_num_cpus
+        if maximum_model_num_gpus is not None and maximum_model_num_gpus < num_gpus:
+            num_gpus = maximum_model_num_gpus
+
         assert system_num_cpus >= num_cpus
         assert system_num_gpus >= num_gpus
 

autogluon/core/version.py

@@ -1,4 +1,4 @@
 """This is the autogluon version file."""
 
-__version__ = "1.2.1b20250417"
+__version__ = "1.2.1b20250419"
 __lite__ = False

{autogluon.core-1.2.1b20250417.dist-info → autogluon.core-1.2.1b20250419.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.core
-Version: 1.2.1b20250417
+Version: 1.2.1b20250419
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community
@@ -45,12 +45,12 @@ Requires-Dist: tqdm<5,>=4.38
 Requires-Dist: requests
 Requires-Dist: matplotlib<3.11,>=3.7.0
 Requires-Dist: boto3<2,>=1.10
-Requires-Dist: autogluon.common==1.2.1b20250417
+Requires-Dist: autogluon.common==1.2.1b20250419
 Provides-Extra: all
-Requires-Dist: hyperopt<0.2.8,>=0.2.7; extra == "all"
 Requires-Dist: ray[default,tune]<2.45,>=2.10.0; extra == "all"
 Requires-Dist: ray[default]<2.45,>=2.10.0; extra == "all"
 Requires-Dist: pyarrow>=15.0.0; extra == "all"
+Requires-Dist: hyperopt<0.2.8,>=0.2.7; extra == "all"
 Provides-Extra: ray
 Requires-Dist: ray[default]<2.45,>=2.10.0; extra == "ray"
 Provides-Extra: raytune
@@ -59,9 +59,9 @@ Requires-Dist: ray[default,tune]<2.45,>=2.10.0; extra == "raytune"
 Requires-Dist: hyperopt<0.2.8,>=0.2.7; extra == "raytune"
 Provides-Extra: tests
 Requires-Dist: pytest; extra == "tests"
-Requires-Dist: types-setuptools; extra == "tests"
 Requires-Dist: types-requests; extra == "tests"
 Requires-Dist: flake8; extra == "tests"
+Requires-Dist: types-setuptools; extra == "tests"
 Requires-Dist: pytest-mypy; extra == "tests"
 
 

{autogluon.core-1.2.1b20250417.dist-info → autogluon.core-1.2.1b20250419.dist-info}/RECORD

@@ -1,9 +1,9 @@
-autogluon.core-1.2.1b20250417-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
+autogluon.core-1.2.1b20250419-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
 autogluon/core/__init__.py,sha256=8KfvvHzXX3a4q6z43Dw1yE7VtbAoiSMaglVpKDy6Xeg,245
 autogluon/core/_setup_utils.py,sha256=TnDboM66MzSp0CLACo_vcNNwhORQWE8JL0kt1IYa7dI,7014
 autogluon/core/constants.py,sha256=nEVLdSFJ-5O-tz3jUD3qPX65RMp7g8qOR38XlurbP4Y,3403
 autogluon/core/problem_type.py,sha256=XJmMgeNBgS7u43pDK-spTivatPyh_INOXveEXwQt-Rw,2993
-autogluon/core/version.py,sha256=9DNociU3CkjmMks3rFL-2UfsFfw26T7_cq6VAom-ic4,91
+autogluon/core/version.py,sha256=KnhYQ6jLbqIvCiI4AYYVbzisJ31dHOaw-tUjtnfNV8M,91
 autogluon/core/augmentation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/core/augmentation/distill_utils.py,sha256=JBlp2WOMNKoJv8aKVwJVRQSalSk8jx36HM7-k_VvkhY,9404
 autogluon/core/calibrate/__init__.py,sha256=eU6qLj7DKUhaz2HHNHDrfroRaLM-mhuSncK_v1UP4F8,62
@@ -40,7 +40,7 @@ autogluon/core/models/__init__.py,sha256=raRAG4wml-xvCKp3zxthda7of9NEWcpafgpvZWJ
 autogluon/core/models/_utils.py,sha256=qswE9n1ge1AJSExgstEbrZiMFmMRa4Mf5Sz8D9-XU6c,2091
 autogluon/core/models/abstract/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/core/models/abstract/_tags.py,sha256=Qr_3an0ZMig24S3OwISa-nTFfWHQe3pwPTiXq4zlEec,3409
-autogluon/core/models/abstract/abstract_model.py,sha256=S4S_cYO_CZtk2tnk6-IwpXP0V7RfWYHsR9R7DmGMnPU,124310
+autogluon/core/models/abstract/abstract_model.py,sha256=y2Qfn6WFcZ5P-KdEzcKFCbbY-3g8-4-yoDfjMzLS6zc,124796
 autogluon/core/models/abstract/abstract_nn_model.py,sha256=IId0ivO8uVvmpnK9OiM2CtPVrP1ewOaQQKtQUDtK7_k,4818
 autogluon/core/models/abstract/model_trial.py,sha256=PKEo1jfLSBCOLM42QE5VBD1u41MaVMRk31zhNhLiqTw,5035
 autogluon/core/models/dummy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -89,11 +89,11 @@ autogluon/core/utils/utils.py,sha256=FMa9kIUAxA3IIBbATmBnNEVObSAivehZ2_zCy3PRR-c
 autogluon/core/utils/version_utils.py,sha256=5-r8hLRKTaZbj5qo2uzE_2E4casH49Ye3WyeHlgHuz4,3252
 autogluon/core/utils/loaders/__init__.py,sha256=W5FAdQvpDcn_uisqJrlSAObWVta-YjJLKGN3NCbEgIo,109
 autogluon/core/utils/savers/__init__.py,sha256=bGWciSxAkj6u06vOC4pTvr22f_1ey0glgvmjCMEOm78,89
-autogluon.core-1.2.1b20250417.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
-autogluon.core-1.2.1b20250417.dist-info/METADATA,sha256=WFLhQI8Kh5SJ8AdlKK6wvNcZrFZlMqmTjsJ8gtqGe6s,12399
-autogluon.core-1.2.1b20250417.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
-autogluon.core-1.2.1b20250417.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-autogluon.core-1.2.1b20250417.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.core-1.2.1b20250417.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.core-1.2.1b20250417.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-autogluon.core-1.2.1b20250417.dist-info/RECORD,,
+autogluon.core-1.2.1b20250419.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
+autogluon.core-1.2.1b20250419.dist-info/METADATA,sha256=v3K3QXBs358LSCUSNzyRbyBKu2L2rWIS_IthaYJejcI,12399
+autogluon.core-1.2.1b20250419.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
+autogluon.core-1.2.1b20250419.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+autogluon.core-1.2.1b20250419.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.core-1.2.1b20250419.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.core-1.2.1b20250419.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+autogluon.core-1.2.1b20250419.dist-info/RECORD,,