autogluon.core 1.2.1b20250205__py3-none-any.whl → 1.2.1b20250207__py3-none-any.whl

autogluon/core/_setup_utils.py

@@ -32,6 +32,7 @@ DEPENDENT_PACKAGES = {
     "async_timeout": ">=4.0,<6",  # Major version cap
     "transformers[sentencepiece]": ">=4.38.0,<5",
     "accelerate": ">=0.34.0,<1.0",
+    "typing-extensions": ">=4.0,<5",
 }
 if LITE_MODE:
     DEPENDENT_PACKAGES = {package: version for package, version in DEPENDENT_PACKAGES.items() if package not in ["psutil", "Pillow", "timm"]}

autogluon/core/models/abstract/abstract_model.py

@@ -96,11 +96,11 @@ class AbstractModel:
 
     def __init__(
         self,
-        path: str = None,
-        name: str = None,
-        problem_type: str = None,
-        eval_metric: Union[str, metrics.Scorer] = None,
-        hyperparameters: dict = None,
+        path: str | None = None,
+        name: str | None = None,
+        problem_type: str | None = None,
+        eval_metric: str | metrics.Scorer | None = None,
+        hyperparameters: dict | None = None,
     ):
         if name is None:
             self.name = self.__class__.__name__
@@ -1176,7 +1176,7 @@ class AbstractModel:
             quantile_levels=self.quantile_levels,
         )
 
-    def save(self, path: str = None, verbose: bool = True) -> str:
+    def save(self, path: str | None = None, verbose: bool = True) -> str:
         """
         Saves the model to disk.