Ensure that custom baseline or reform triggers import of Reform class (…

…#1997) * feat: Ensure that custom baseline or reform triggers Reform input in getHeaderCode * chore: Lint --------- Co-authored-by: PolicyEngine[bot] <[email protected]>
PolicyEngine · Sep 2, 2024 · 5a48767 · 5a48767
1 parent 7166ed4
commit 5a48767
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/src/data/reformDefinitionCode.js b/src/data/reformDefinitionCode.js
@@ -39,8 +39,11 @@ export function getHeaderCode(type, metadata, policy) {
     lines.push("from " + metadata.package + " import Microsimulation");
   }
 
-  // If there is a reform, add the following Python imports
-  if (Object.keys(policy.reform.data).length > 0) {
+  // If either baseline or reform is custom, add the following Python imports
+  if (
+    Object.keys(policy.reform.data).length > 0 ||
+    Object.keys(policy.baseline.data).length > 0
+  ) {
     lines.push("from policyengine_core.reforms import Reform");
   }