Fix pylint error: numpy concatenate has no dtype.

PiperOrigin-RevId: 706708506
google-research · Dec 16, 2024 · 109129c · 109129c
1 parent 9fcde1e
commit 109129c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/protoscribe/corpus/tools/quantize_sketches_simple_main.py b/protoscribe/corpus/tools/quantize_sketches_simple_main.py
@@ -156,8 +156,8 @@ def _collect_points(examples: Iterable[dict[str, Any]]) -> Array:
     touch_points.append(sketch[list(pen_touch_ids), :2])
     lift_points.append(sketch[list(pen_lift_ids), :2])
 
-  touch_points = np.concatenate(touch_points, dtype=np.float32)
-  lift_points = np.concatenate(lift_points, dtype=np.float32)
+  touch_points = np.concatenate(touch_points).astype(dtype=np.float32)
+  lift_points = np.concatenate(lift_points).astype(dtype=np.float32)
   num_touch = touch_points.shape[0]
   num_lift = lift_points.shape[0]
   logging.info(