kedro-org · SajidAlamQB · Oct 25, 2023 · Oct 25, 2023 · Oct 25, 2023 · Oct 25, 2023
diff --git a/...cutter.repo_name }}/src/tests/__init__.py → ...okiecutter.repo_name }}/tests/__init__.py b/...cutter.repo_name }}/src/tests/__init__.py → ...okiecutter.repo_name }}/tests/__init__.py
diff --git a/...o_name }}/src/tests/pipelines/__init__.py → ....repo_name }}/tests/pipelines/__init__.py b/...o_name }}/src/tests/pipelines/__init__.py → ....repo_name }}/tests/pipelines/__init__.py
diff --git a/...cutter.repo_name }}/src/tests/test_run.py → ...okiecutter.repo_name }}/tests/test_run.py b/...cutter.repo_name }}/src/tests/test_run.py → ...okiecutter.repo_name }}/tests/test_run.py
diff --git a/...cutter.repo_name }}/src/tests/__init__.py → ...okiecutter.repo_name }}/tests/__init__.py b/...cutter.repo_name }}/src/tests/__init__.py → ...okiecutter.repo_name }}/tests/__init__.py
diff --git a/...r.repo_name }}/src/tests/test_pipeline.py → ...utter.repo_name }}/tests/test_pipeline.py b/...r.repo_name }}/src/tests/test_pipeline.py → ...utter.repo_name }}/tests/test_pipeline.py
diff --git a/...cutter.repo_name }}/src/tests/test_run.py → ...okiecutter.repo_name }}/tests/test_run.py b/...cutter.repo_name }}/src/tests/test_run.py → ...okiecutter.repo_name }}/tests/test_run.py
diff --git a/...cutter.repo_name }}/src/tests/__init__.py → ...okiecutter.repo_name }}/tests/__init__.py b/...cutter.repo_name }}/src/tests/__init__.py → ...okiecutter.repo_name }}/tests/__init__.py
diff --git a/...r.repo_name }}/src/tests/test_pipeline.py → ...utter.repo_name }}/tests/test_pipeline.py b/...r.repo_name }}/src/tests/test_pipeline.py → ...utter.repo_name }}/tests/test_pipeline.py
diff --git a/...cutter.repo_name }}/src/tests/test_run.py → ...okiecutter.repo_name }}/tests/test_run.py b/...cutter.repo_name }}/src/tests/test_run.py → ...okiecutter.repo_name }}/tests/test_run.py
diff --git a/...cutter.repo_name }}/src/tests/__init__.py → ...okiecutter.repo_name }}/tests/__init__.py b/...cutter.repo_name }}/src/tests/__init__.py → ...okiecutter.repo_name }}/tests/__init__.py
diff --git a/...r.repo_name }}/src/tests/test_pipeline.py → ...utter.repo_name }}/tests/test_pipeline.py b/...r.repo_name }}/src/tests/test_pipeline.py → ...utter.repo_name }}/tests/test_pipeline.py
diff --git a/...cutter.repo_name }}/src/tests/test_run.py → ...okiecutter.repo_name }}/tests/test_run.py b/...cutter.repo_name }}/src/tests/test_run.py → ...okiecutter.repo_name }}/tests/test_run.py
diff --git a/...cutter.repo_name }}/src/tests/__init__.py → ...okiecutter.repo_name }}/tests/__init__.py b/...cutter.repo_name }}/src/tests/__init__.py → ...okiecutter.repo_name }}/tests/__init__.py
diff --git a/...o_name }}/src/tests/pipelines/__init__.py → ....repo_name }}/tests/pipelines/__init__.py b/...o_name }}/src/tests/pipelines/__init__.py → ....repo_name }}/tests/pipelines/__init__.py
diff --git a/...cutter.repo_name }}/src/tests/test_run.py → ...okiecutter.repo_name }}/tests/test_run.py b/...cutter.repo_name }}/src/tests/test_run.py → ...okiecutter.repo_name }}/tests/test_run.py
diff --git a/...cutter.repo_name }}/src/tests/__init__.py → ...okiecutter.repo_name }}/tests/__init__.py b/...cutter.repo_name }}/src/tests/__init__.py → ...okiecutter.repo_name }}/tests/__init__.py
diff --git a/...o_name }}/src/tests/pipelines/__init__.py → ....repo_name }}/tests/pipelines/__init__.py b/...o_name }}/src/tests/pipelines/__init__.py → ....repo_name }}/tests/pipelines/__init__.py
diff --git a/.../src/tests/pipelines/test_data_science.py → ...e }}/tests/pipelines/test_data_science.py b/.../src/tests/pipelines/test_data_science.py → ...e }}/tests/pipelines/test_data_science.py
diff --git a/...cutter.repo_name }}/src/tests/test_run.py → ...okiecutter.repo_name }}/tests/test_run.py b/...cutter.repo_name }}/src/tests/test_run.py → ...okiecutter.repo_name }}/tests/test_run.py
diff --git a/...cutter.repo_name }}/src/tests/__init__.py → spaceflights-pandas/tests/__init__.py b/...cutter.repo_name }}/src/tests/__init__.py → spaceflights-pandas/tests/__init__.py
diff --git a/...o_name }}/src/tests/pipelines/__init__.py → ...lights-pandas/tests/pipelines/__init__.py b/...o_name }}/src/tests/pipelines/__init__.py → ...lights-pandas/tests/pipelines/__init__.py
diff --git a/...cutter.repo_name }}/src/tests/test_run.py → spaceflights-pandas/tests/test_run.py b/...cutter.repo_name }}/src/tests/test_run.py → spaceflights-pandas/tests/test_run.py
diff --git a/...cutter.repo_name }}/src/tests/__init__.py → ...okiecutter.repo_name }}/tests/__init__.py b/...cutter.repo_name }}/src/tests/__init__.py → ...okiecutter.repo_name }}/tests/__init__.py
diff --git a/...o_name }}/src/tests/pipelines/__init__.py → ....repo_name }}/tests/pipelines/__init__.py b/...o_name }}/src/tests/pipelines/__init__.py → ....repo_name }}/tests/pipelines/__init__.py
diff --git a/.../src/tests/pipelines/test_data_science.py → ...e }}/tests/pipelines/test_data_science.py b/.../src/tests/pipelines/test_data_science.py → ...e }}/tests/pipelines/test_data_science.py
diff --git a/...cutter.repo_name }}/src/tests/test_run.py → ...okiecutter.repo_name }}/tests/test_run.py b/...cutter.repo_name }}/src/tests/test_run.py → ...okiecutter.repo_name }}/tests/test_run.py
diff --git a/spaceflights-pyspark/hooks/post_gen_project.py b/spaceflights-pyspark/hooks/post_gen_project.py
@@ -8,14 +8,14 @@
 from kedro.framework.cli.starters import _parse_add_ons_input
 
 
-def main(selected_add_ons_list):
+def main(selected_add_ons):
     current_dir = Path.cwd()
     requirements_file_path = current_dir / "requirements.txt"
     pyproject_file_path = current_dir / "pyproject.toml"
     python_package_name = '{{ cookiecutter.python_package }}'
 
     # Handle template directories and requirements according to selected add-ons
-    setup_template_add_ons(selected_add_ons_list, requirements_file_path, pyproject_file_path, python_package_name)
+    setup_template_add_ons(selected_add_ons, requirements_file_path, pyproject_file_path, python_package_name)
 
     # Sort requirements.txt file in alphabetical order
     sort_requirements(requirements_file_path)
@@ -24,10 +24,8 @@ def main(selected_add_ons_list):
 if __name__ == "__main__":
     # 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)
 
-    # Execute the script only if the PySpark add-on (represented by "6") is selected.
+    # Execute the script only if the Pyspark add-on is selected.
     # This ensures the script doesn't run with kedro new --starter but only with the add-ons flow option.
-    if "6" in selected_add_ons_list:
-        main(selected_add_ons_list)
+    if "Pyspark" in selected_add_ons:
+        main(selected_add_ons)
diff --git a/...cutter.repo_name }}/src/tests/__init__.py → ...okiecutter.repo_name }}/tests/__init__.py b/...cutter.repo_name }}/src/tests/__init__.py → ...okiecutter.repo_name }}/tests/__init__.py
diff --git a/...o_name }}/src/tests/pipelines/__init__.py → ....repo_name }}/tests/pipelines/__init__.py b/...o_name }}/src/tests/pipelines/__init__.py → ....repo_name }}/tests/pipelines/__init__.py
diff --git a/.../src/tests/pipelines/test_data_science.py → ...e }}/tests/pipelines/test_data_science.py b/.../src/tests/pipelines/test_data_science.py → ...e }}/tests/pipelines/test_data_science.py
diff --git a/...cutter.repo_name }}/src/tests/test_run.py → ...okiecutter.repo_name }}/tests/test_run.py b/...cutter.repo_name }}/src/tests/test_run.py → ...okiecutter.repo_name }}/tests/test_run.py