Use extension instead of platform

rnett · rnett · commit 28e6ba7ced51 · 2021-07-01T10:41:44.000-07:00
Signed-off-by: Ryan Nett &lt;JNett96@gmail.com&gt;
diff --git a/tensorflow-kotlin-parent/tensorflow-kotlin-jupyter/src/main/kotlin/org/tensorflow/jupyter/TensorflowKotlinIntegration.kt b/tensorflow-kotlin-parent/tensorflow-kotlin-jupyter/src/main/kotlin/org/tensorflow/jupyter/TensorflowKotlinIntegration.kt
@@ -44,8 +44,7 @@ public class TensorflowKotlinIntegration : JupyterIntegration() {
                       "No version property found in the $tensorflowPropertiesFile resource, did you overwrite it?")
             }
 
-    // TODO use ext instead of platform https://github.com/Kotlin/kotlin-jupyter/issues/285
-    dependencies("org.tensorflow:tensorflow-core-platform-gpu:$version")
+    dependencies("org.tensorflow:tensorflow-core-api:jar:$ext:$version")
     dependencies("org.tensorflow:tensorflow-core-kotlin-jupyter:$version")
   }
 }

Original file line number	Diff line number	Diff line change
`@@ -44,8 +44,7 @@ public class TensorflowKotlinIntegration : JupyterIntegration() {`
`44`	`44`	`"No version property found in the $tensorflowPropertiesFile resource, did you overwrite it?")`
`45`	`45`	`}`
`46`	`46`
`47`		`- // TODO use ext instead of platform https://github.com/Kotlin/kotlin-jupyter/issues/285`
`48`		`- dependencies("org.tensorflow:tensorflow-core-platform-gpu:$version")`
	`47`	`+ dependencies("org.tensorflow:tensorflow-core-api:jar:$ext:$version")`
`49`	`48`	`dependencies("org.tensorflow:tensorflow-core-kotlin-jupyter:$version")`
`50`	`49`	`}`
`51`	`50`	`}`