Fix spaceflights-pyspark starter (#168)

* Update post_gen_project.py Signed-off-by: SajidAlamQB <[email protected]> * Update post_gen_project.py Signed-off-by: SajidAlamQB <[email protected]> --------- Signed-off-by: SajidAlamQB <[email protected]>
kedro-org · Oct 25, 2023 · 2c5274b · 2c5274b
1 parent 7e78565
commit 2c5274b
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/spaceflights-pyspark/hooks/post_gen_project.py b/spaceflights-pyspark/hooks/post_gen_project.py
@@ -1,10 +1,11 @@
 from pathlib import Path
 
 from kedro.templates.project.hooks.utils import (
-    parse_add_ons_input,
+
     setup_template_add_ons,
     sort_requirements,
 )
+from kedro.framework.cli.starters import _parse_add_ons_input
 
 
 def main(selected_add_ons_list):
@@ -24,7 +25,7 @@ def main(selected_add_ons_list):
     # Get the selected add-ons from cookiecutter
     selected_add_ons = "{{ cookiecutter.add_ons }}"
     # Parse the add-ons to get a list
-    selected_add_ons_list = parse_add_ons_input(selected_add_ons)
+    selected_add_ons_list = _parse_add_ons_input(selected_add_ons)
 
     # Execute the script only if the PySpark add-on (represented by "6") is selected.
     # This ensures the script doesn't run with kedro new --starter but only with the add-ons flow option.