Fix pyspark example in docs (#1452)

Signed-off-by: SajidAlamQB <90610031+SajidAlamQB@users.noreply.github.com>
kedro-org · Apr 20, 2022 · 1b15589 · 1b15589
1 parent e80451d
commit 1b15589
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/docs/source/tools_integration/pyspark.md b/docs/source/tools_integration/pyspark.md
@@ -38,10 +38,13 @@ class CustomContext(KedroContext):
         package_name: str,
         project_path: Union[Path, str],
         config_loader: ConfigLoader,
+        hook_manager: PluginManager,
         env: str = None,
         extra_params: Dict[str, Any] = None,
     ):
-        super().__init__(package_name, project_path, config_loader, env, extra_params)
+        super().__init__(
+            package_name, project_path, config_loader, hook_manager, env, extra_params
+        )
         self.init_spark_session()
 
     def init_spark_session(self) -> None:
@@ -53,7 +56,7 @@ class CustomContext(KedroContext):
 
         # Initialise the spark session
         spark_session_conf = (
-            SparkSession.builder.appName(self.package_name)
+            SparkSession.builder.appName(self._package_name)
             .enableHiveSupport()
             .config(conf=spark_conf)
         )