[NumPy] Remove references to deprecated NumPy type aliases.

This change replaces references to a number of deprecated NumPy type aliases (np.bool, np.int, np.float, np.complex, np.object, np.str) with their recommended replacement (bool, int, float, complex, object, str).

NumPy 1.24 drops the deprecated aliases, so we must remove uses before updating NumPy.

PiperOrigin-RevId: 497189015

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,

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]],