add paths in tasks.py

Author: sarahmish

README.md

@@ -67,19 +67,7 @@ In this example we use `gpt_detector` pipeline and set some hyperparameters. In
 In addtion, the `SigLLM` object takes in a `decimal` argument to determine how many digits from the float value include. Here, we don't want to keep any decimal values, so we set it to zero.
 
 ```python3
-import os
 from sigllm import SigLLM
-from mlblocks.discovery import get_pipelines_paths, add_pipelines_path
-from mlblocks.discovery import get_primitives_paths, add_primitives_path
-
-print(get_pipelines_paths())
-path = os.path.join(os.getcwd(), '..', '..', 'sigllm', 'pipelines', 'detector')
-add_pipelines_path(path)
-print(get_pipelines_paths())
-
-path = os.path.join(os.getcwd(), '..', '..', 'sigllm', 'primitives', 'jsons')
-add_primitives_path(path)
-
 
 hyperparameters = {
     "orion.primitives.timeseries_anomalies.find_anomalies#1": {

tasks.py

@@ -10,6 +10,7 @@
 from packaging.requirements import Requirement
 from packaging.version import Version
 
+from mlblocks.discovery import add_primitives_path, add_pipelines_path
 
 def _get_minimum_versions(dependencies, python_version):
     min_versions = {}
@@ -69,8 +70,21 @@ def minimum(c):
     unit(c)
 
 
+def add_paths():
+    base_path = os.path.join(os.getcwd(), 'sigllm')
+    primitives_path = os.path.join(base_path, 'primitives', 'jsons')
+    pipeline_paths = [
+        os.path.join(base_path, 'pipelines', 'detector'),
+        os.path.join(base_path, 'pipelines', 'prompter')
+    ]
+    add_primitives_path(primitives_path)
+    for pipeline_path in pipeline_paths:
+        add_pipelines_path(pipeline_path)
+
+
 @task
 def readme(c):
+    add_paths()
     test_path = Path('tests/readme_test')
     if test_path.exists() and test_path.is_dir():
         shutil.rmtree(test_path)