policyengine-core 3.32.2__tar.gz → 3.32.3__tar.gz

{policyengine_core-3.32.2 → policyengine_core-3.32.3}/CHANGELOG.md

@@ -1,3 +1,10 @@
+## [3.32.3] - 2026-09-04
+
+### Fixed
+
+- Correct the public type annotations for methods that add, replace, and update variables.
+
+
 ## [3.32.2] - 2026-09-04
 
 ### Fixed

{policyengine_core-3.32.2 → policyengine_core-3.32.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: policyengine-core
-Version: 3.32.2
+Version: 3.32.3
 Summary: Core microsimulation engine enabling country-specific policy models.
 Project-URL: Homepage, https://github.com/policyengine/policyengine-core
 Author-email: PolicyEngine <hello@policyengine.org>

{policyengine_core-3.32.2 → policyengine_core-3.32.3}/policyengine_core/taxbenefitsystems/tax_benefit_system.py

@@ -219,11 +219,11 @@ class TaxBenefitSystem:
 
         return variable
 
-    def add_variable(self, variable: Variable) -> Variable:
+    def add_variable(self, variable: Type[Variable]) -> Variable:
         """Adds an OpenFisca variable to the tax and benefit system.
 
         Args:
-            variable: The variable to add. Must be a subclass of Variable.
+            variable: The variable class to add. Must be a subclass of Variable.
 
         Raises:
             policyengine_core.errors.VariableNameConflictError: if a variable with the same name have previously been added to the tax and benefit system.
@@ -232,7 +232,7 @@ class TaxBenefitSystem:
         self.data_modified = True
         return self.load_variable(variable, update=False)
 
-    def replace_variable(self, variable: str) -> Variable:
+    def replace_variable(self, variable: Type[Variable]) -> None:
         """
         Replaces an existing OpenFisca variable in the tax and benefit system by a new one.
 
@@ -240,7 +240,7 @@ class TaxBenefitSystem:
 
         If no variable with the given name exists in the tax and benefit system, no error will be raised and the new variable will be simply added.
 
-        :param Variable variable: New variable to add. Must be a subclass of Variable.
+        :param Variable variable: New variable class to add. Must be a subclass of Variable.
         """
         name = variable.__name__
         if self.variables.get(name) is not None:
@@ -248,7 +248,7 @@ class TaxBenefitSystem:
         self.load_variable(variable, update=False)
         self.data_modified = True
 
-    def update_variable(self, variable: str) -> Variable:
+    def update_variable(self, variable: Type[Variable]) -> Variable:
         """
         Updates an existing OpenFisca variable in the tax and benefit system.
 
@@ -258,7 +258,7 @@ class TaxBenefitSystem:
 
         If no variable with the given name exists in the tax and benefit system, no error will be raised and the new variable will be simply added.
 
-        :param Variable variable: Variable to add. Must be a subclass of Variable.
+        :param Variable variable: Variable class to add. Must be a subclass of Variable.
         """
         self.data_modified = True
         return self.load_variable(variable, update=True)
@@ -377,9 +377,9 @@ class TaxBenefitSystem:
         for subdirectory in subdirectories:
             self.add_variables_from_directory(subdirectory)
 
-    def add_variables(self, *variables: List[Type[Variable]]):
+    def add_variables(self, *variables: Type[Variable]) -> None:
         """
-        Adds a list of OpenFisca Variables to the `TaxBenefitSystem`.
+        Adds OpenFisca Variable classes to the `TaxBenefitSystem`.
 
         See also :any:`add_variable`
         """

{policyengine_core-3.32.2 → policyengine_core-3.32.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "policyengine-core"
-version = "3.32.2"
+version = "3.32.3"
 description = "Core microsimulation engine enabling country-specific policy models."
 readme = "README.md"
 authors = [