Internal change

PiperOrigin-RevId: 497719172
google-research · Dec 26, 2022 · 16e47ac · 16e47ac
1 parent c28c6fa
commit 16e47ac
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/vmoe/evaluate/fewshot_test.py b/vmoe/evaluate/fewshot_test.py
@@ -106,7 +106,7 @@ def setUp(self):
     valid = np.concatenate([
         np.ones((3, 16)),
         np.concatenate([np.ones((1, 12)), np.zeros((1, 4))], axis=1),
-    ], axis=0).astype(np.bool)
+    ], axis=0).astype(bool)
     dataset = tf.data.Dataset.from_tensor_slices({
         'image': images,
         'label': labels,

diff --git a/vmoe/moe_test.py b/vmoe/moe_test.py
@@ -251,7 +251,7 @@ def test_top_items_per_expert_einsum(self):
         dispatcher.dispatch_weights,
         np.asarray([[[0, 1], [0, 0], [0, 0]],
                     [[1, 0], [0, 1], [0, 1]],
-                    [[0, 0], [1, 0], [1, 0]]], dtype=np.bool))
+                    [[0, 0], [1, 0], [1, 0]]], dtype=bool))
     np.testing.assert_array_almost_equal(
         dispatcher.combine_weights,
         np.asarray([[[.0, .4], [.0, .0], [.0, .0]],