Use extension instead of platform

Signed-off-by: Ryan Nett <[email protected]>
tensorflow · Jul 1, 2021 · 28e6ba7 · 28e6ba7
1 parent 8293765
commit 28e6ba7
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/...flow-kotlin-jupyter/src/main/kotlin/org/tensorflow/jupyter/TensorflowKotlinIntegration.kt b/...flow-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")
   }
 }