From fcf3b6695566ff7b6163a5573137d3ac28e3e32e Mon Sep 17 00:00:00 2001
From: bculkin2442 <bjculkin@mix.wvu.edu>
Date: Mon, 21 Mar 2016 21:42:46 -0400
Subject: Minor polishing of things

---
 BJC-Utils2/src/main/java/bjc/utils/parserutils/AST.java | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

(limited to 'BJC-Utils2/src/main/java/bjc/utils/parserutils/AST.java')

diff --git a/BJC-Utils2/src/main/java/bjc/utils/parserutils/AST.java b/BJC-Utils2/src/main/java/bjc/utils/parserutils/AST.java
index 7b93cc9..153fc41 100644
--- a/BJC-Utils2/src/main/java/bjc/utils/parserutils/AST.java
+++ b/BJC-Utils2/src/main/java/bjc/utils/parserutils/AST.java
@@ -88,7 +88,7 @@ public class AST<T> {
 	 * @return The collapsed value of the tree
 	 */
 	public <E, T2> E collapse(Function<T, T2> tokenTransform,
-			Map<T, BinaryOperator<T2>> nodeTransform,
+			Function<T, BinaryOperator<T2>> nodeTransform,
 			Function<T2, E> resultTransform) {
 		return resultTransform
 				.apply(internalCollapse(tokenTransform, nodeTransform));
@@ -98,7 +98,7 @@ public class AST<T> {
 	 * Internal recursive collapser
 	 */
 	private <T2> T2 internalCollapse(Function<T, T2> tokenTransform,
-			Map<T, BinaryOperator<T2>> nodeTransform) {
+			Function<T, BinaryOperator<T2>> nodeTransform) {
 		if (left == null && right == null) {
 			return tokenTransform.apply(token);
 		} else {
@@ -107,7 +107,7 @@ public class AST<T> {
 			T2 rightCollapsed = right.internalCollapse(tokenTransform,
 					nodeTransform);
 
-			return nodeTransform.get(token).apply(leftCollapsed,
+			return nodeTransform.apply(token).apply(leftCollapsed,
 					rightCollapsed);
 		}
 	}
-- 
cgit v1.2.3