Added policy parametrization from kwargs (#301)

* Added policy parametrization from kwargs

* Added policy parametrization from kwargs

* bump version

---------

Co-authored-by: Matthew Russo <mdrusso@mit.edu>

Author: vitaglianog

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "palimpzest"
-version = "1.5.2"
+version = "1.5.3"
 description = "Palimpzest is a system which enables anyone to process AI-powered analytical queries simply by defining them in a declarative language"
 readme = "README.md"
 requires-python = ">=3.12"

src/palimpzest/core/data/dataset.py

@@ -716,6 +716,7 @@ def optimize_and_run(self, config: QueryProcessorConfig | None = None, train_dat
         policy = construct_policy_from_kwargs(**kwargs)
         if policy is not None:
             kwargs["policy"] = policy
+            config.policy = policy
 
         # construct unique logical op ids for all operators in this dataset
         self._generate_unique_logical_op_ids()