Adjust tests in test_process_model.py

OpenSourceEconomics · Feb 6, 2024 · b7c561c · b7c561c
1 parent 0fabc75
commit b7c561c
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/tests/test_process_model.py b/tests/test_process_model.py
@@ -217,7 +217,10 @@ def raw_func(health, wealth):  # noqa: ARG001
 
     raw_func._stochastic_info = StochasticInfo()
 
-    variable_info = pd.DataFrame({"is_state": [True, True]}, index=["health", "wealth"])
+    variable_info = pd.DataFrame(
+        {"is_discrete": [True, True]},
+        index=["health", "wealth"],
+    )
 
     grids = {
         "health": jnp.array([1, 4]),
@@ -245,7 +248,7 @@ def raw_func(health, wealth):  # noqa: ARG001
     raw_func._stochastic_info = StochasticInfo()
 
     variable_info = pd.DataFrame(
-        {"is_state": [False, True]},
+        {"is_discrete": [False, True]},
         index=["health", "wealth"],
     )