fix test

pyg-team · Jul 13, 2022 · 022edec · 022edec
1 parent 31a2099
commit 022edec
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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