UAST: register udf for sql queries

src-d · Sep 19, 2017 · f31b1e5 · f31b1e5
1 parent 6ff11c5
commit f31b1e5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/main/scala/tech/sourced/api/Implicits.scala b/src/main/scala/tech/sourced/api/Implicits.scala
@@ -45,6 +45,7 @@ object Implicits {
       Implicits.checkCols(df, "path", "content")
       import org.apache.spark.sql.functions.udf
       val extractUastUDF = udf(BblfshUDF.extractUAST)
+      df.sparkSession.udf.register("extract_uasts", extractUastUDF)
       df.withColumn("uast", extractUastUDF('path, 'content))
     }