ODP-2117: Added additional java imports for pyspark

acceldata-io · Aug 23, 2024 · c3c7f80 · c3c7f80
1 parent 212e1e0
commit c3c7f80
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/repl/src/main/resources/fake_shell.py b/repl/src/main/resources/fake_shell.py
@@ -589,8 +589,12 @@ def main():
             java_import(gateway.jvm, "org.apache.spark.SparkConf")
             java_import(gateway.jvm, "org.apache.spark.api.java.*")
             java_import(gateway.jvm, "org.apache.spark.api.python.*")
+            java_import(gateway.jvm, "org.apache.spark.ml.python.*")
             java_import(gateway.jvm, "org.apache.spark.mllib.api.python.*")
+            java_import(gateway.jvm, "org.apache.spark.resource.*")
+
             java_import(gateway.jvm, "org.apache.spark.sql.*")
+            java_import(gateway.jvm, "org.apache.spark.sql.api.python.*")
             java_import(gateway.jvm, "org.apache.spark.sql.hive.*")
             java_import(gateway.jvm, "scala.Tuple2")