autogluon.common 1.2.1b20250217__py3-none-any.whl → 1.2.1b20250219__py3-none-any.whl

autogluon/common/utils/resource_utils.py

@@ -35,11 +35,14 @@ class ResourceManager:
         return num_gpus
 
     @staticmethod
-    def get_gpu_count_torch():
+    def get_gpu_count_torch() -> int:
         try:
             import torch
 
-            num_gpus = torch.cuda.device_count()
+            if not torch.cuda.is_available():
+                num_gpus = 0
+            else:
+                num_gpus = torch.cuda.device_count()
         except Exception:
             num_gpus = 0
         return num_gpus

autogluon/common/utils/try_import.py

@@ -31,7 +31,7 @@ def try_import_mxboard():
 
 
 def try_import_ray() -> ModuleType:
-    RAY_MAX_VERSION = "2.41.0"  # sync with core/setup.py
+    RAY_MAX_VERSION = "2.43.0"  # sync with core/setup.py
     ray_max_version_os_map = dict(
         Darwin=RAY_MAX_VERSION,
         Windows=RAY_MAX_VERSION,

autogluon/common/utils/utils.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import logging
 import os
 import platform
@@ -22,7 +24,9 @@ logger = logging.getLogger(__name__)
 LITE_MODE: bool = __lite__ is not None and __lite__
 
 
-def setup_outputdir(path, warn_if_exist=True, create_dir=True, path_suffix=None):
+def setup_outputdir(
+    path: str | None, warn_if_exist: bool = True, create_dir: bool = True, path_suffix: str | None = None
+) -> str:
     is_s3_path = False
     if path:
         assert isinstance(path, (str, Path)), (
@@ -41,7 +45,9 @@ def setup_outputdir(path, warn_if_exist=True, create_dir=True, path_suffix=None)
     else:
         utcnow = datetime.now(timezone.utc)
         timestamp = utcnow.strftime("%Y%m%d_%H%M%S")
-        path = os.path.join("AutogluonModels", f"ag-{timestamp}{path_suffix}")
+        path = os.path.join("AutogluonModels", f"ag-{timestamp}")
+        if path_suffix:
+            path = os.path.join(path, path_suffix)
         for i in range(1, 1000):
             try:
                 if create_dir:
@@ -52,10 +58,13 @@ def setup_outputdir(path, warn_if_exist=True, create_dir=True, path_suffix=None)
                         raise FileExistsError
                     break
             except FileExistsError:
-                path = os.path.join("AutogluonModels", f"ag-{timestamp}-{i:03d}{path_suffix}")
+                path = os.path.join("AutogluonModels", f"ag-{timestamp}-{i:03d}")
+                if path_suffix:
+                    path = os.path.join(path, path_suffix)
         else:
             raise RuntimeError("more than 1000 jobs launched in the same second")
         logger.log(25, f'No path specified. Models will be saved in: "{path}"')
+        warn_if_exist = False  # Don't warn about the folder existing since we just created it
 
     if warn_if_exist and not is_s3_path:
         try:

autogluon/common/version.py

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

{autogluon.common-1.2.1b20250217.dist-info → autogluon.common-1.2.1b20250219.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.common
-Version: 1.2.1b20250217
+Version: 1.2.1b20250219
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community
@@ -42,10 +42,10 @@ Requires-Dist: boto3<2,>=1.10
 Requires-Dist: psutil<7.0.0,>=5.7.3
 Requires-Dist: tqdm<5,>=4.38
 Provides-Extra: tests
-Requires-Dist: types-setuptools; extra == "tests"
 Requires-Dist: pytest-mypy; extra == "tests"
 Requires-Dist: pytest; extra == "tests"
 Requires-Dist: types-requests; extra == "tests"
+Requires-Dist: types-setuptools; extra == "tests"
 
 
 

{autogluon.common-1.2.1b20250217.dist-info → autogluon.common-1.2.1b20250219.dist-info}/RECORD

@@ -1,8 +1,8 @@
-autogluon.common-1.2.1b20250217-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
+autogluon.common-1.2.1b20250219-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
 autogluon/common/__init__.py,sha256=Q0tQ4UOtUEhPpj83k_xzQ9bmtoTSvfg_3WgPi1QopJk,352
 autogluon/common/dataset.py,sha256=slfU8CxGFNJ86_2wKkT_zPMPr-iPB-sz7klz0YqJECo,1580
 autogluon/common/space.py,sha256=oRSG-uFmsLACT42mT2vRp3NSuV8MaQz6X4sYwcZruxs,5457
-autogluon/common/version.py,sha256=KtJNzyEXgs-zqexx_OthJT01Df0QPzQ_fCAJ0rRSse0,91
+autogluon/common/version.py,sha256=-6B9LRx5ypsiEngHly5V3KaGydGIcWNulylOlKPbR2U,91
 autogluon/common/features/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/common/features/feature_metadata.py,sha256=GDvZXyaNkDeuNv8omO4K7tkCPmMdpJzD5rH_hKtZD9E,25783
 autogluon/common/features/infer_types.py,sha256=MkfVDoY__0Rr__zsDSKszmvYMq5RQRmbNboTG_tm64o,6492
@@ -37,17 +37,17 @@ autogluon/common/utils/multiprocessing_utils.py,sha256=gTbOMnwOrMYaY-NJy1GVS5aIv
 autogluon/common/utils/nvutil.py,sha256=tVyCdUDoM7RSkO1Ek01NVRFL8bbFnQBryezYpN-BX1M,4804
 autogluon/common/utils/pandas_utils.py,sha256=sirUUQkKZsmMwmXWEUZJ1n3ZYFQWqwz4iu-tAhZdv68,2432
 autogluon/common/utils/path_converter.py,sha256=OO0Jr5vhcqrEvZrNqem5tmgC6FZA3_O8sPylQUKUgKs,2441
-autogluon/common/utils/resource_utils.py,sha256=ZP6KGh8HliacadptvftL8x4dm_cw0Hwm27BNpN_dTnI,8352
+autogluon/common/utils/resource_utils.py,sha256=CmU8OsaXoreja43gMGRZCg8BiKQdHvDxWW0y7auZE2Q,8456
 autogluon/common/utils/s3_utils.py,sha256=Tj2VyUGUc7Vrx5KkCGKapvFQyaVvWpJNpjssfXg2Sl4,17142
 autogluon/common/utils/simulation_utils.py,sha256=iEZcRWzP8xLQ8racIfOGC5V_A0W_xNgxPkGuBfX6hh8,3573
 autogluon/common/utils/system_info.py,sha256=9BkFcdq1co0EMnZkk_AYdhxSEP3RqcU6uIp-iYx9bdc,4479
-autogluon/common/utils/try_import.py,sha256=FVHRwyeUCyJyUEm3GevcTjOeatpIt38AxGSJLaDSvZo,7499
-autogluon/common/utils/utils.py,sha256=dxb-OAXBLLw6mPtj0mr4WZbLEJlin8wChZ4hH5LhQ8U,7581
-autogluon.common-1.2.1b20250217.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
-autogluon.common-1.2.1b20250217.dist-info/METADATA,sha256=7XSO8m0UxVPu7BYoIUyhTsd3wY4rPhsS-UBmLtdirGo,11658
-autogluon.common-1.2.1b20250217.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
-autogluon.common-1.2.1b20250217.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-autogluon.common-1.2.1b20250217.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.common-1.2.1b20250217.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.common-1.2.1b20250217.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-autogluon.common-1.2.1b20250217.dist-info/RECORD,,
+autogluon/common/utils/try_import.py,sha256=k_d9RvHs_6ShL5gN7OuVtNHhZZ9BOxz4t6piOkfpug4,7499
+autogluon/common/utils/utils.py,sha256=pdGu5kIZi3UlmdDgIQg6HsPVJIhBT__ZYb2imP6_NZg,7907
+autogluon.common-1.2.1b20250219.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
+autogluon.common-1.2.1b20250219.dist-info/METADATA,sha256=5ft2f34XwHD7WgsaIl1MHzQbLWp-7vWmBvX7IOMpmHg,11658
+autogluon.common-1.2.1b20250219.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
+autogluon.common-1.2.1b20250219.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+autogluon.common-1.2.1b20250219.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.common-1.2.1b20250219.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.common-1.2.1b20250219.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+autogluon.common-1.2.1b20250219.dist-info/RECORD,,