nextrec 0.4.24__py3-none-any.whl → 0.4.27__py3-none-any.whl

nextrec/__version__.py

@@ -1 +1 @@
-__version__ = "0.4.24"
+__version__ = "0.4.27"

nextrec/basic/asserts.py

@@ -0,0 +1,72 @@
+"""
+Assert function definitions for NextRec models.
+
+Date: create on 01/01/2026
+Checkpoint: edit on 01/01/2026
+Author: Yang Zhou, zyaztec@gmail.com
+"""
+
+from __future__ import annotations
+
+from nextrec.utils.types import TaskTypeName, TrainingModeName
+
+
+def assert_task(
+    task: list[TaskTypeName] | TaskTypeName | None,
+    nums_task: int,
+    *,
+    model_name: str,
+) -> None:
+    if task is None:
+        raise ValueError(f"{model_name} requires task to be specified.")
+
+    # case 1: task is str
+    if isinstance(task, str):
+        if nums_task != 1:
+            raise ValueError(
+                f"{model_name} received task='{task}' but nums_task={nums_task}. "
+                "String task is only allowed for single-task models."
+            )
+        return  # single-task, valid
+
+    # case 2: task is list
+    if not isinstance(task, list):
+        raise TypeError(
+            f"{model_name} requires task to be a string or a list of strings."
+        )
+
+    # list but length == 1
+    if len(task) == 1:
+        if nums_task != 1:
+            raise ValueError(
+                f"{model_name} received task list of length 1 but nums_task={nums_task}. "
+                "Length-1 task list is only allowed for single-task models."
+            )
+        return  # single-task, valid
+
+    # multi-task: length must match nums_task
+    if len(task) != nums_task:
+        raise ValueError(
+            f"{model_name} requires task length {nums_task}, got {len(task)}."
+        )
+
+
+def assert_training_mode(
+    training_mode: TrainingModeName | list[TrainingModeName],
+    nums_task: int,
+    *,
+    model_name: str,
+) -> None:
+    valid_modes = {"pointwise", "pairwise", "listwise"}
+    if not isinstance(training_mode, list):
+        raise TypeError(
+            f"[{model_name}-init Error] training_mode must be a list with length {nums_task}."
+        )
+    if len(training_mode) != nums_task:
+        raise ValueError(
+            f"[{model_name}-init Error] training_mode list length must match number of tasks."
+        )
+    if any(mode not in valid_modes for mode in training_mode):
+        raise ValueError(
+            f"[{model_name}-init Error] training_mode must be one of {'pointwise', 'pairwise', 'listwise'}."
+        )

nextrec/basic/loggers.py

@@ -2,7 +2,7 @@
 NextRec Basic Loggers
 
 Date: create on 27/10/2025
-Checkpoint: edit on 27/12/2025
+Checkpoint: edit on 01/01/2026
 Author: Yang Zhou, zyaztec@gmail.com
 """
 
@@ -190,6 +190,19 @@ class BasicLogger:
     def close(self) -> None:
         for backend in self.backends:
             backend.close()
+        for backend in self.backends:
+            if isinstance(backend, SwanLabLogger):
+                swanlab = backend.swanlab
+                if not backend.enabled or swanlab is None:
+                    continue
+                finish_fn = getattr(swanlab, "finish", None)
+                if finish_fn is None:
+                    continue
+                try:
+                    finish_fn()
+                except TypeError:
+                    finish_fn()
+                break
 
 
 class TensorBoardLogger(MetricsLoggerBackend):
@@ -369,10 +382,14 @@ class TrainingLogger(BasicLogger):
         wandb_kwargs = dict(wandb_kwargs or {})
         wandb_kwargs.setdefault("config", {})
         wandb_kwargs["config"].update(config)
+        if "notes" in wandb_kwargs:
+            wandb_kwargs["config"].pop("note", None)
 
         swanlab_kwargs = dict(swanlab_kwargs or {})
         swanlab_kwargs.setdefault("config", {})
         swanlab_kwargs["config"].update(config)
+        if "description" in swanlab_kwargs:
+            swanlab_kwargs["config"].pop("note", None)
 
         self.wandb_logger = None
         if use_wandb: