autogluon.common 1.2.1b20250218__tar.gz → 1.2.1b20250220__tar.gz

{autogluon.common-1.2.1b20250218 → autogluon.common-1.2.1b20250220}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.common
-Version: 1.2.1b20250218
+Version: 1.2.1b20250220
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community

{autogluon.common-1.2.1b20250218 → autogluon.common-1.2.1b20250220}/src/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-1.2.1b20250218 → autogluon.common-1.2.1b20250220}/src/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-1.2.1b20250218 → autogluon.common-1.2.1b20250220}/src/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-1.2.1b20250218 → autogluon.common-1.2.1b20250220}/src/autogluon/common/version.py

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

{autogluon.common-1.2.1b20250218 → autogluon.common-1.2.1b20250220}/src/autogluon.common.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.common
-Version: 1.2.1b20250218
+Version: 1.2.1b20250220
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community

{autogluon.common-1.2.1b20250218 → autogluon.common-1.2.1b20250220}/src/autogluon.common.egg-info/requires.txt

@@ -6,6 +6,6 @@ tqdm<5,>=4.38
 
 [tests]
 types-setuptools
+pytest
 types-requests
 pytest-mypy
-pytest