[Bugfix] Fix actnn tests

cogdl/utils/utils.py

@@ -25,6 +25,12 @@ def build_args_from_dict(dic):
     return args
 
 
+def update_args_from_dict(args, dic):
+    for key, value in dic.items():
+        args.__setattr__(key, value)
+    return args
+
+
 def set_random_seed(seed):
     random.seed(seed)
     np.random.seed(seed)

tests/models/ssl/test_generative_models.py

@@ -42,6 +42,7 @@ def get_default_args():
         "eval_agc": False,
         "subgraph_sampling": False,
         "sample_size": 128,
+        "actnn": False,
     }
     args = build_args_from_dict(args)
     dataset = build_dataset(args)

tests/tasks/test_link_prediction.py

@@ -1,10 +1,7 @@
 import torch
 
-from torch import argsort
-from cogdl import options
-from cogdl.tasks import build_task, register_task
+from cogdl.tasks import build_task
 from cogdl.datasets import build_dataset
-from cogdl.models import build_model
 from cogdl.utils import build_args_from_dict
 
 
@@ -21,6 +18,7 @@ def get_default_args():
         "activation": "relu",
         "residual": False,
         "norm": None,
+        "actnn": False,
     }
     return build_args_from_dict(default_dict)
 
@@ -59,6 +57,7 @@ def get_kg_default_args():
         "activation": "relu",
         "residual": False,
         "norm": None,
+        "actnn": False,
     }
     return build_args_from_dict(default_dict)
 
@@ -129,6 +128,7 @@ def get_kge_default_args():
         "checkpoint": False,
         "save_dir": ".",
         "device_id": [0],
+        "actnn": False,
     }
     return build_args_from_dict(default_dict)
 
@@ -193,6 +193,7 @@ def get_gnn_link_prediction_args():
         "activation": "relu",
         "residual": False,
         "norm": None,
+        "actnn": False,
     }
     return build_args_from_dict(args)