ci test

SciSharp · Nov 10, 2023 · 09d466d · 09d466d
1 parent 2cb5fd6
commit 09d466d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/test/TensorFlowNET.UnitTest/Training/BasicLinearModel.cs b/test/TensorFlowNET.UnitTest/Training/BasicLinearModel.cs
@@ -15,7 +15,9 @@ public class BasicLinearModel
         [TestMethod]
         public void LinearRegression()
         {
-            tf.Graph().as_default();
+            var graph = tf.Graph().as_default();
+            var sess = new Session(graph);
+            sess.as_default();
 
             // Initialize the weights to `5.0` and the bias to `0.0`
             // In practice, these should be initialized to random values (for example, with `tf.random.normal`)