change test file

c091f12a · 张彦钊 · 330993c8 · c091f12a
Commit c091f12a authored Mar 19, 2019 by 张彦钊
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

test.py tensnsorflow/test.py +1 -1

No files found.
--- a/tensnsorflow/test.py
+++ b/tensnsorflow/test.py
@@ -26,7 +26,7 @@ def esmm_pre():
    gw = SparkContext._gateway

    # Import TiExtensions
-    java_import(gw.jvm, "org.apache.spark.sql.TiExtensions")
+    java_import(gw.jvm, "org.apache.spark.sql.TiContext")

    # Inject TiExtensions, and get a TiContext
    ti = gw.jvm.TiExtensions.getInstance(spark._jsparkSession).getOrCreateTiContext(spark._jsparkSession)