Fix session.close for local test (#573)

alibaba · Jul 21, 2021 · ec4caff · ec4caff
1 parent e84765f
commit ec4caff
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python/tests/test_run_locally.py b/python/tests/test_run_locally.py
@@ -270,6 +270,7 @@ def test_demo_lazy_mode(ogbn_mag_small):
         "edge_type": "cites",
     }
     train(config, r[1])
+    sess.close()
 
 
 @pytest.mark.skipif(sys.platform == "darwin", reason="Mac no need to run this test.")