From 022edec1ddbb439ffff5547918432e4d175e705e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Julian=20F=C3=B6rster?= <fjulian@users.noreply.github.com>
Date: Wed, 13 Jul 2022 19:45:45 +0200
Subject: [PATCH] fix test

---
 test/graphgym/test_graphgym.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test/graphgym/test_graphgym.py b/test/graphgym/test_graphgym.py
index 1a0991d7605b..d8fcb3e09b31 100644
--- a/test/graphgym/test_graphgym.py
+++ b/test/graphgym/test_graphgym.py
@@ -136,7 +136,7 @@ def test_graphgym_module(tmpdir):
     assert isinstance(scheduler[0], torch.optim.lr_scheduler.CosineAnnealingLR)
 
     cfg.params = params_count(model)
-    assert cfg.params == 23880
+    assert cfg.params == 23883
 
     keys = {"loss", "true", "pred_score", "step_end_time"}
     # test training step