Fix numpy 2 compatibility

PiperOrigin-RevId: 725567326

Author: Conchylicultor

gemma/__init__.py

@@ -16,4 +16,4 @@
 
 # A new PyPI release will be pushed every time `__version__` is increased.
 # When changing this, also update the CHANGELOG.md.
-__version__ = '2.0.5'
+__version__ = '2.0.6'

gemma/gm/data/_functional.py

@@ -99,12 +99,12 @@ def make_next_token_prediction_fields(
     The input, target and mask, all of length `prompt_len + response_len - 1`.
   """
   # Concatenate the prompt and response tokens.
-  sequence = np.concat([prompt, response])
+  sequence = np.concatenate([prompt, response])
 
   # Create the loss mask.
-  target_mask = np.concat([
-      np.zeros((len(prompt) - 1,), dtype=np.bool),
-      np.ones((len(response),), dtype=np.bool),
+  target_mask = np.concatenate([
+      np.zeros((len(prompt) - 1,), dtype=np.bool_),
+      np.ones((len(response),), dtype=np.bool_),
   ])
 
   return NextTokenPredictionFields(