From a7b074875f241d98877de51a93717d56cf2d19e4 Mon Sep 17 00:00:00 2001
From: Carlo Zancanaro <carlo@zancanaro.id.au>
Date: Mon, 6 Jul 2015 22:46:02 +1000
Subject: Fix the warnings from the previous commit

---
 src/main/java/au/id/zancanaro/javacheck/ShrinkTree.java | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

(limited to 'src/main/java/au/id/zancanaro/javacheck/ShrinkTree.java')

diff --git a/src/main/java/au/id/zancanaro/javacheck/ShrinkTree.java b/src/main/java/au/id/zancanaro/javacheck/ShrinkTree.java
index 6e21f27..ddef800 100644
--- a/src/main/java/au/id/zancanaro/javacheck/ShrinkTree.java
+++ b/src/main/java/au/id/zancanaro/javacheck/ShrinkTree.java
@@ -56,17 +56,17 @@ public class ShrinkTree<T> {
     public static <T> Stream<ShrinkTree<T>[]> promoteChildren(ShrinkTree<? super T>[] trees) {
         return IntStream.range(0, trees.length)
                 .mapToObj(index -> trees[index].getChildren().map(child ->
-                        IntStream.range(0, trees.length)
+                        (ShrinkTree<T>[]) IntStream.range(0, trees.length)
                                 .mapToObj(i -> (i == index ? child : trees[i]))
                                 .toArray(ShrinkTree[]::new)))
-                .flatMap(x -> x)
-                .map(x -> (ShrinkTree<T>[]) x);
+                .flatMap(x -> x);
     }
 
+    @SuppressWarnings("unchecked")
     public static <T> Stream<ShrinkTree<T>[]> removeChildren(ShrinkTree<? super T>[] trees) {
         return IntStream.range(0, trees.length)
                 .mapToObj(index ->
-                        IntStream.range(0, trees.length)
+                        (ShrinkTree<T>[]) IntStream.range(0, trees.length)
                                 .filter(i -> i != index)
                                 .mapToObj(i -> trees[i])
                                 .toArray(ShrinkTree[]::new));
-- 
cgit v1.2.3