nextrec 0.5.3__py3-none-any.whl → 0.5.5__py3-none-any.whl

nextrec/__version__.py

@@ -1 +1 @@
-__version__ = "0.5.3"
+__version__ = "0.5.5"

nextrec/basic/model.py

@@ -2099,6 +2099,7 @@ class BaseModel(SummarySet, FeatureSet, nn.Module):
         ]
 
         ctx = mp.get_context("spawn")
+        error_queue = ctx.SimpleQueue()
         processes = []
         for rank in range(num_processes):
             process = ctx.Process(
@@ -2115,6 +2116,7 @@ class BaseModel(SummarySet, FeatureSet, nn.Module):
                     processor,
                     rank,
                     num_processes,
+                    error_queue,
                 ),
             )
             process.start()
@@ -2127,8 +2129,18 @@ class BaseModel(SummarySet, FeatureSet, nn.Module):
 
         for process in processes:
             if process.exitcode not in (0, None):
+                errors = []
+                try:
+                    while not error_queue.empty():
+                        errors.append(error_queue.get_nowait())
+                except Exception:
+                    pass
+                error_text = (
+                    "\n\n".join(errors) if errors else "No worker traceback captured."
+                )
                 raise RuntimeError(
-                    "[BaseModel-predict-streaming Error] One or more inference processes failed."
+                    "[BaseModel-predict-streaming Error] One or more inference processes failed.\n"
+                    + error_text
                 )
         # Merge part files
         existing_parts = [p for p in part_paths if p.exists()]
@@ -2827,22 +2839,32 @@ def predict_streaming_worker(
     processor: Any | None,
     shard_rank: int,
     shard_count: int,
+    error_queue: "mp.SimpleQueue[str] | None" = None,
 ) -> None:
-    model.eval()
-    model.predict_streaming(
-        data=data_path,
-        batch_size=batch_size,
-        save_path=save_path,
-        save_format=save_format,
-        stream_chunk_size=stream_chunk_size,
-        return_dataframe=False,
-        num_workers=num_workers,
-        prefetch_factor=prefetch_factor,
-        processor=processor,
-        num_processes=1,
-        shard_rank=shard_rank,
-        shard_count=shard_count,
-    )
+    try:
+        model.eval()
+        model.predict_streaming(
+            data=data_path,
+            batch_size=batch_size,
+            save_path=save_path,
+            save_format=save_format,
+            stream_chunk_size=stream_chunk_size,
+            return_dataframe=False,
+            num_workers=num_workers,
+            prefetch_factor=prefetch_factor,
+            processor=processor,
+            num_processes=1,
+            shard_rank=shard_rank,
+            shard_count=shard_count,
+        )
+    except Exception:
+        if error_queue is not None:
+            import traceback
+
+            error_queue.put(
+                f"[PredictWorker Error] rank={shard_rank}\n{traceback.format_exc()}"
+            )
+        raise
 
 
 class BaseMatchModel(BaseModel):

nextrec/cli.py

@@ -25,7 +25,7 @@ import resource
 import sys
 import time
 from pathlib import Path
-from typing import Any, Dict, List
+from typing import Any, Dict
 
 import pandas as pd
 
@@ -137,7 +137,7 @@ def train_model(train_config_path: str) -> None:
     model_cfg = read_yaml(model_cfg_path)
 
     # Extract id_column from data config for GAUC metrics
-    id_column = data_cfg.get("id_column") or data_cfg.get("user_id_column")
+    id_column = data_cfg.get("id_column")
     id_columns = [id_column] if id_column else []
 
     log_cli_section("Data")
@@ -378,6 +378,7 @@ def train_model(train_config_path: str) -> None:
         sparse_features,
         sequence_features,
         target,
+        id_columns,
         device,
     )
 
@@ -585,19 +586,16 @@ def predict_model(predict_config_path: str) -> None:
         sparse_features=sparse_features,
         sequence_features=sequence_features,
         target=target_cols,
+        id_columns=id_columns,
         device=predict_cfg.get("device", "cpu"),
     )
-    model.id_columns = id_columns
+
     model.load_model(
         model_file, map_location=predict_cfg.get("device", "cpu"), verbose=True
     )
 
-    id_columns = []
-    if predict_cfg.get("id_column"):
-        id_columns = [predict_cfg["id_column"]]
-        model.id_columns = id_columns
-
     effective_id_columns = id_columns or model.id_columns
+
     log_cli_section("Features")
     log_kv_lines(
         [

nextrec/utils/config.py

@@ -388,6 +388,7 @@ def build_model_instance(
     sparse_features: List["SparseFeature"],
     sequence_features: List["SequenceFeature"],
     target: List[str],
+    id_columns: list[str] | str | None,
     device: str,
 ) -> Any:
     """
@@ -400,6 +401,7 @@ def build_model_instance(
         sparse_features: List of sparse feature objects
         sequence_features: List of sequence feature objects
         target: List of target column names
+        id_columns: Identifier column name(s) for GAUC or ID passthrough
         device: Device string (e.g., 'cpu', 'cuda:0')
     """
     dense_map = {f.name: f for f in dense_features}
@@ -436,7 +438,7 @@ def build_model_instance(
         param.kind == inspect.Parameter.VAR_KEYWORD for param in sig_params.values()
     )
 
-    init_kwargs: Dict[str, Any] = dict(params_cfg)
+    init_kwargs = dict(params_cfg)
 
     # Explicit bindings (model_config.feature_bindings) take priority
     for param_name, binding in feature_bindings_cfg.items():
@@ -517,6 +519,8 @@ def build_model_instance(
 
     if accepts("target"):
         init_kwargs.setdefault("target", target)
+    if accepts("id_columns") or accepts_var_kwargs:
+        init_kwargs.setdefault("id_columns", id_columns)
     if accepts("device"):
         init_kwargs.setdefault("device", device)
 

{nextrec-0.5.3.dist-info → nextrec-0.5.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nextrec
-Version: 0.5.3
+Version: 0.5.5
 Summary: A comprehensive recommendation library with match, ranking, and multi-task learning models
 Project-URL: Homepage, https://github.com/zerolovesea/NextRec
 Project-URL: Repository, https://github.com/zerolovesea/NextRec
@@ -73,7 +73,7 @@ Description-Content-Type: text/markdown
 ![Python](https://img.shields.io/badge/Python-3.10+-blue.svg)
 ![PyTorch](https://img.shields.io/badge/PyTorch-1.10+-ee4c2c.svg)
 ![License](https://img.shields.io/badge/License-Apache%202.0-green.svg)
-![Version](https://img.shields.io/badge/Version-0.5.3-orange.svg)
+![Version](https://img.shields.io/badge/Version-0.5.5-orange.svg)
 [![Ask DeepWiki](https://deepwiki.com/badge.svg)](https://deepwiki.com/zerolovesea/NextRec)
 
 中文文档 | [English Version](README_en.md)
@@ -259,11 +259,11 @@ nextrec --mode=predict --predict_config=path/to/predict_config.yaml
 
 预测结果固定保存到 `{checkpoint_path}/predictions/{name}.{save_data_format}`。
 
-> 截止当前版本0.5.3，NextRec CLI支持单机训练，分布式训练相关功能尚在开发中。
+> 截止当前版本0.5.5，NextRec CLI支持单机训练，分布式训练相关功能尚在开发中。
 
 ## 兼容平台
 
-当前最新版本为0.5.3，所有模型和测试代码均已在以下平台通过验证，如果开发者在使用中遇到兼容问题，请在issue区提出错误报告及系统版本：
+当前最新版本为0.5.5，所有模型和测试代码均已在以下平台通过验证，如果开发者在使用中遇到兼容问题，请在issue区提出错误报告及系统版本：
 
 | 平台 | 配置 | 
 |------|------|

{nextrec-0.5.3.dist-info → nextrec-0.5.5.dist-info}/RECORD

@@ -1,6 +1,6 @@
 nextrec/__init__.py,sha256=_M3oUqyuvQ5k8Th_3wId6hQ_caclh7M5ad51XN09m98,235
-nextrec/__version__.py,sha256=tgzuqHKcEdKBaP57F5oXxq4XlW2n9J4Fj8ZGu7nGOZg,22
-nextrec/cli.py,sha256=ddYlk2zXF7iVFpONhG9Ofwb9dVgXfy112nk_JOHy1kI,29840
+nextrec/__version__.py,sha256=78mfpLewKVki6c9UONSUdlVme_JsN9ZwIfp4Hf4jmG0,22
+nextrec/cli.py,sha256=ywGOMJ-iVLH6bOk2VX_n0r530V11xbFUHVwu6-AlzPw,29675
 nextrec/basic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nextrec/basic/activation.py,sha256=rU-W-DHgiD3AZnMGmD014ChxklfP9BpedDTiwtdgXhA,2762
 nextrec/basic/asserts.py,sha256=eaB4FZJ7Sbh9S8PLJZNiYd7Zi98ca2rDi4S7wSYCaEw,1473
@@ -10,7 +10,7 @@ nextrec/basic/heads.py,sha256=WqvavaH6Y8Au8dLaoUfH2AaOOWgYvjZI5US8avkQNsQ,4009
 nextrec/basic/layers.py,sha256=tawggQMMHlYTGpnubxUAvDPDJe_Lpq-HpLLCSjbJV54,37320
 nextrec/basic/loggers.py,sha256=jJUzUt_kMpjpV2Mqr7qBENWA1olEutTI7dFnpmndUUw,13845
 nextrec/basic/metrics.py,sha256=nVz3AkKwsxj_M97CoZWyQj8_Y9ZM_Icvw_QCM6c33Bc,26262
-nextrec/basic/model.py,sha256=YkGLo9iwU1dWZXI198t4u5Ncm35uymhSe61oAToQc3U,135049
+nextrec/basic/model.py,sha256=J8fkwQF2Y5YUajqucfSpsNuz8sbLqsE3mupCfbkZDhc,135869
 nextrec/basic/session.py,sha256=mrIsjRJhmvcAfoO1pXX-KB3SK5CCgz89wH8XDoAiGEI,4475
 nextrec/basic/summary.py,sha256=r5d8VtxJsgY6WRAVRa2-UcGb_sYA9VjMKDfQ31928qM,20494
 nextrec/data/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -71,7 +71,7 @@ nextrec/models/tree_base/catboost.py,sha256=hXINyx7iianwDxOZx3SLm0i-YP1jiC3HcAeq
 nextrec/models/tree_base/lightgbm.py,sha256=VilMU7SgfHR5LAaaoQo-tY1vkzpSvWovIrgaSeuJ1-A,2263
 nextrec/models/tree_base/xgboost.py,sha256=thOmDIC_nitno_k2mcH2cj2VcS07f9veTG01FMOO-28,1957
 nextrec/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nextrec/utils/config.py,sha256=Ngd_u8ZS5br4lIqrBJ_ecLquMF4KJi6TPAGqLZg8H4s,20485
+nextrec/utils/config.py,sha256=uChqdxA3X6gn2G8tlUrobZ1AbaxdSQMt87eocSQVgog,20691
 nextrec/utils/console.py,sha256=RnSUplJnyanSQ6TyMQkP7S1j2rGMver1DbFVqNH6_1k,13581
 nextrec/utils/data.py,sha256=JN30npycFMo_caBQJ5815tciRW3r7YDqSQxnJuXhlVA,22299
 nextrec/utils/embedding.py,sha256=akAEc062MG2cD7VIOllHaqtwzAirQR2gq5iW7oKpGAU,1449
@@ -80,8 +80,8 @@ nextrec/utils/model.py,sha256=K1-ZfS1umSTVo-lNHKNkBb3xvWnfiJJSTRicSVDpk4s,5148
 nextrec/utils/onnx_utils.py,sha256=KIVV_ELYzj3kCswfsSBZ1F2OnSwRJnXj7sxDBwBoBaA,8668
 nextrec/utils/torch_utils.py,sha256=_a9e6GXa3QKuu0E5RL44QRZ1iJSobbtNcPB3vtaCsu8,12313
 nextrec/utils/types.py,sha256=LFwYCBRo5WeYUh5LSCuyP1Lg9ez0Ih00Es3fUttGAFw,2273
-nextrec-0.5.3.dist-info/METADATA,sha256=TAIvFtcvAhHJ8PPc5YdbJM6ZBgRUnd3dMkKlLXR3Oj4,23464
-nextrec-0.5.3.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-nextrec-0.5.3.dist-info/entry_points.txt,sha256=NN-dNSdfMRTv86bNXM7d3ZEPW2BQC6bRi7QP7i9cIps,45
-nextrec-0.5.3.dist-info/licenses/LICENSE,sha256=COP1BsqnEUwdx6GCkMjxOo5v3pUe4-Go_CdmQmSfYXM,1064
-nextrec-0.5.3.dist-info/RECORD,,
+nextrec-0.5.5.dist-info/METADATA,sha256=XbDqdm6X1jgRyk_GSMXyMVKtZe1AsuCNTNjcC4NTD-A,23464
+nextrec-0.5.5.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+nextrec-0.5.5.dist-info/entry_points.txt,sha256=NN-dNSdfMRTv86bNXM7d3ZEPW2BQC6bRi7QP7i9cIps,45
+nextrec-0.5.5.dist-info/licenses/LICENSE,sha256=COP1BsqnEUwdx6GCkMjxOo5v3pUe4-Go_CdmQmSfYXM,1064
+nextrec-0.5.5.dist-info/RECORD,,