Fix weight constraint test.

2023-04-19 10:29:38 -07:00 · 2023-04-19 10:29:38 -07:00 · bbcd6eee8a
commit bbcd6eee8a
parent aae25f518f
1 changed files with 1 additions and 0 deletions
--- a/keras_core/constraints/constraints_test.py
+++ b/keras_core/constraints/constraints_test.py
@ -29,6 +29,7 @@ class ConstraintsTest(testing.TestCase):
    def test_non_neg(self):
        constraint_fn = constraints.NonNeg()
        output = constraint_fn(get_example_array())
+        output = np.array(output)
        self.assertTrue((np.min(output, axis=1) >= 0.0).all())

    def test_unit_norm(self):