Fix type of MaterializedResult.getOnlyColumnAsSet

prestodb · Dec 21, 2017 · a7fa6da · a7fa6da
1 parent 4ce94e0
commit a7fa6da
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/presto-main/src/main/java/com/facebook/presto/testing/MaterializedResult.java b/presto-main/src/main/java/com/facebook/presto/testing/MaterializedResult.java
@@ -189,11 +189,11 @@ public String toString()
                 .toString();
     }
 
-    public Set<String> getOnlyColumnAsSet()
+    public Set<Object> getOnlyColumnAsSet()
     {
         checkState(types.size() == 1, "result set must have exactly one column");
         return rows.stream()
-                .map(row -> (String) row.getField(0))
+                .map(row -> row.getField(0))
                 .collect(toImmutableSet());
     }