From 5912673b0d1cc9c0b9c90241619978eb0484aea3 Mon Sep 17 00:00:00 2001
From: Michael Armbrust <michael@databricks.com>
Date: Fri, 25 Mar 2016 11:46:36 -0700
Subject: [PATCH] Fix JoinPerformance compilation

Author: Michael Armbrust <michael@databricks.com>

Closes #55 from marmbrus/fixJoinPerf.
---
 src/main/scala/com/databricks/spark/sql/perf/Benchmark.scala | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/main/scala/com/databricks/spark/sql/perf/Benchmark.scala b/src/main/scala/com/databricks/spark/sql/perf/Benchmark.scala
index 436b8bf..ccb8960 100644
--- a/src/main/scala/com/databricks/spark/sql/perf/Benchmark.scala
+++ b/src/main/scala/com/databricks/spark/sql/perf/Benchmark.scala
@@ -24,7 +24,7 @@ import scala.language.implicitConversions
 import scala.util.Try
 
 import org.apache.spark.rdd.RDD
-import org.apache.spark.sql.{DataFrame, SQLContext}
+import org.apache.spark.sql.{Dataset, DataFrame, SQLContext}
 import org.apache.spark.sql.catalyst.analysis.UnresolvedRelation
 import org.apache.spark.SparkContext
 
@@ -355,7 +355,7 @@ abstract class Benchmark(
 
   case class Table(
       name: String,
-      data: DataFrame)
+      data: Dataset[_])
 
   import reflect.runtime._, universe._
   import reflect.runtime._