From aa65bdd1fdb2ca7bbd37d66d68780ef4f5e69091 Mon Sep 17 00:00:00 2001
From: Beini <xbn20@mails.tsinghua.edu.cn>
Date: Mon, 13 Dec 2021 03:45:10 +0000
Subject: [PATCH] dataset

---
 autogl/module/model/dgl/hetero/han.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/autogl/module/model/dgl/hetero/han.py b/autogl/module/model/dgl/hetero/han.py
index bbf2bf9..b527ede 100644
--- a/autogl/module/model/dgl/hetero/han.py
+++ b/autogl/module/model/dgl/hetero/han.py
@@ -103,6 +103,7 @@ class HAN(nn.Module):
                     "meta_paths",
                     "num_features",
                     "num_class",
+                    "out_key",
                     "num_layers",
                     "hidden",
                     "heads",