do not rely on implicit

spotify · Feb 10, 2020 · dbfb1d9 · dbfb1d9
1 parent 53f5420
commit dbfb1d9
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/contrib/flo-scio_2.12/src/main/scala/com/spotify/flo/contrib/scio/ScioOperator.scala b/contrib/flo-scio_2.12/src/main/scala/com/spotify/flo/contrib/scio/ScioOperator.scala
@@ -117,21 +117,21 @@ class ScioOperator[T] extends TaskOperator[ScioJobSpec.Provider[T], ScioJobSpec[
     spec.pipeline(sc)
 
     // Start job
-    val scioResult = Try(sc.run())
-    scioResult match {
+    val sec = Try(sc.run())
+    sec match {
       case Failure(t) => return spec.failure(t)
       case _ =>
     }
 
     // Report job id
-    scioResult.get.internal match {
+    sec.get.pipelineResult match {
       case job: DataflowPipelineJob => reportDataflowJob(spec.taskId, job, listener)
       case _ =>
     }
 
     // Wait for job to complete
-    val done = Try(scioResult.get.waitUntilDone())
-    done match {
+    val scioResult = sec.map(_.waitUntilDone())
+    scioResult match {
       case Failure(t) => return spec.failure(t)
       case _ =>
     }