clean up

ucr-riple · Oct 3, 2024 · 7ae149e · 7ae149e
1 parent 1ab923f
commit 7ae149e
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/annotator-core/src/main/java/edu/ucr/cs/riple/core/injectors/AnnotationInjector.java b/annotator-core/src/main/java/edu/ucr/cs/riple/core/injectors/AnnotationInjector.java
@@ -29,9 +29,7 @@
 import edu.ucr.cs.riple.injector.changes.AddAnnotation;
 import edu.ucr.cs.riple.injector.changes.RemoveAnnotation;
 import java.util.Set;
-import java.util.function.Function;
 import java.util.stream.Collectors;
-import java.util.stream.Stream;
 
 /** Wrapper tool used to inject annotations to the source code. */
 public abstract class AnnotationInjector {
@@ -53,9 +51,7 @@ public void removeFixes(Set<Fix> fixes) {
     }
     Set<RemoveAnnotation> toRemove =
         fixes.stream()
-            .flatMap(
-                (Function<Fix, Stream<RemoveAnnotation>>)
-                    fix -> fix.changes.stream().map(AddAnnotation::getReverse))
+            .flatMap(fix -> fix.changes.stream().map(AddAnnotation::getReverse))
             .collect(Collectors.toSet());
     removeAnnotations(toRemove);
   }