From 1876b2bb4ed3fe0894eddf91d84bcc34912fcb11 Mon Sep 17 00:00:00 2001 From: "Jian Zhang (James)" <6593865@qq.com> Date: Thu, 12 Dec 2024 22:54:17 -0800 Subject: [PATCH] Update examples/network_traffic_prediction/nr_models.py Co-authored-by: xiang song(charlie.song) --- examples/network_traffic_prediction/nr_models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/network_traffic_prediction/nr_models.py b/examples/network_traffic_prediction/nr_models.py index 4d6067ce5..753aded6c 100644 --- a/examples/network_traffic_prediction/nr_models.py +++ b/examples/network_traffic_prediction/nr_models.py @@ -176,7 +176,7 @@ def combine_st_ts_feats(types, st_feats, ts_feats): Both features should be in a dict or a list of dict format, and have the same length. The feature dict is like: {ntype1: tensor, ntype2: tensor, ...} or - [{type1: tensor, type2: tensor, ...}, {type1: tensor, type2: tensor, ...}], which is + [{etype1: tensor, etype2: tensor, ...}, {etype1: tensor, etype2: tensor, ...}], which is for edge features in multiple layers. The major issue is that not all types have the two types of features. So, we