fix: json study_id and then rename to stid

opentargets · Sep 24, 2018 · d14e248 · d14e248
1 parent d18e74f
commit d14e248
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/main/scala/ot/geckopipe/index/V2DIndex.scala b/src/main/scala/ot/geckopipe/index/V2DIndex.scala
@@ -57,7 +57,7 @@ object V2DIndex extends LazyLogging  {
     StructField("rs_id", StringType) :: Nil)
 
   val studiesSchema = StructType(
-    StructField("stid", StringType, false) ::
+    StructField("study_id", StringType, false) ::
       StructField("pmid", StringType) ::
       StructField("pub_date", StringType) ::
       StructField("pub_journal", StringType) ::
@@ -129,6 +129,7 @@ object V2DIndex extends LazyLogging  {
     val removeSuffix = udf((col: String) => col.split("\\.").head)
 
     val pStudies = studies
+      .withColumnRenamed("study_id", "stid")
       .withColumn("trait_efos", when(col("trait_efos").isNotNull,
         split(col("trait_efos"),";")))