Fixed tests

Kotlin · Nov 20, 2020 · 445093d · 445093d
1 parent 418974d
commit 445093d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/api/src/test/kotlin/org/jetbrains/kotlinx/dl/api/core/models/SequentialCompilationTest.kt b/api/src/test/kotlin/org/jetbrains/kotlinx/dl/api/core/models/SequentialCompilationTest.kt
@@ -160,6 +160,7 @@ internal class SequentialModelTest {
                             "Name: Init_dense_2_dense_bias; Type: Fill; Out #tensors:  1\n" +
                             "Name: Assign_dense_2_dense_bias; Type: Assign; Out #tensors:  1\n" +
                             "Name: Placeholder; Type: Placeholder; Out #tensors:  1\n" +
+                            "Name: numberOfLosses; Type: Placeholder; Out #tensors:  1\n" +
                             "Name: Conv2d; Type: Conv2D; Out #tensors:  1\n" +
                             "Name: BiasAdd; Type: BiasAdd; Out #tensors:  1\n" +
                             "Name: Relu; Type: Relu; Out #tensors:  1\n" +
@@ -616,4 +617,4 @@ internal class SequentialModelTest {
         assertEquals(model.layers[2].name, "flatten_3")
         assertEquals(model.layers[3].name, "dense_4")
     }
-}
+}