pin numpy < 2, handle deprecation of bool (#832)

modal-labs · Aug 5, 2024 · 42fbc1e · 42fbc1e
1 parent 958d8e8
commit 42fbc1e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/06_gpu_and_ml/text-to-pokemon/text_to_pokemon/pokemon_naming.py b/06_gpu_and_ml/text-to-pokemon/text_to_pokemon/pokemon_naming.py
@@ -15,7 +15,7 @@
 rnn_image = modal.Image.debian_slim().pip_install(
     "keras",
     "pandas",
-    "numpy",
+    "numpy<2",
     "tensorflow",
 )
 
@@ -176,8 +176,8 @@ def prep_dataset(
             .replace("\n", " ")
         )
 
-    X = np.zeros((num_sequences, max_sequence_len, num_chars), dtype=np.bool)
-    Y = np.zeros((num_sequences, num_chars), dtype=np.bool)
+    X = np.zeros((num_sequences, max_sequence_len, num_chars), dtype=bool)
+    Y = np.zeros((num_sequences, num_chars), dtype=bool)
 
     for i, sequence in enumerate(sequences):
         for j, char in enumerate(sequence):