override tempLocation for Dataflow runner

spotify · Oct 21, 2016 · 93b32b4 · 93b32b4
1 parent 36efa33
commit 93b32b4
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/scio-core/src/main/scala/com/spotify/scio/ScioContext.scala b/scio-core/src/main/scala/com/spotify/scio/ScioContext.scala
@@ -153,6 +153,10 @@ class ScioContext private[scio] (val options: PipelineOptions,
     }
   }
 
+  Try(optionsAs[DataflowPipelineOptions]).foreach { o =>
+    o.setTempLocation(o.getGcpTempLocation)
+  }
+
   {
     val o = optionsAs[ScioOptions]
     o.setScalaVersion(scalaVersion)