From e30388ff6bae311431aff06e4c07347774fb5b62 Mon Sep 17 00:00:00 2001
From: "Joshua E. Jodesty" <joshua.jodesty@gmail.com>
Date: Mon, 18 Feb 2019 14:20:59 -0500
Subject: [PATCH] e-courage: modularizing backwards compatability

---
 cadCAD/configuration/utils/__init__.py            | 7 ++++++-
 cadCAD/configuration/utils/depreciationHandler.py | 1 -
 simulations/validation/config2.py                 | 6 +++---
 3 files changed, 9 insertions(+), 5 deletions(-)

diff --git a/cadCAD/configuration/utils/__init__.py b/cadCAD/configuration/utils/__init__.py
index 2a60d3e..bb7f4d3 100644
--- a/cadCAD/configuration/utils/__init__.py
+++ b/cadCAD/configuration/utils/__init__.py
@@ -4,6 +4,8 @@ from copy import deepcopy
 from fn.func import curried
 import pandas as pd
 
+# Temporary
+from cadCAD.configuration.utils.depreciationHandler import sanitize_partial_state_updates
 from cadCAD.utils import dict_filter, contains_type
 
 
@@ -13,7 +15,10 @@ class TensorFieldReport:
         self.config_proc = config_proc
 
     # ToDo: backwards compatibility
-    def create_tensor_field(self, partial_state_updates, exo_proc, keys=['policies', 'variables']):
+    def create_tensor_field(self, partial_state_updates, exo_proc, keys = ['policies', 'variables']):
+
+        partial_state_updates = sanitize_partial_state_updates(partial_state_updates) # Temporary
+
         dfs = [self.config_proc.create_matrix_field(partial_state_updates, k) for k in keys]
         df = pd.concat(dfs, axis=1)
         for es, i in zip(exo_proc, range(len(exo_proc))):
diff --git a/cadCAD/configuration/utils/depreciationHandler.py b/cadCAD/configuration/utils/depreciationHandler.py
index 1f2862a..b62b192 100644
--- a/cadCAD/configuration/utils/depreciationHandler.py
+++ b/cadCAD/configuration/utils/depreciationHandler.py
@@ -21,7 +21,6 @@ def sanitize_partial_state_updates(partial_state_updates):
     # for backwards compatibility we accept the old keys
     # ('behaviors' and 'states') and rename them
     def rename_keys(d):
-
         if 'behaviors' in d:
             d['policies'] = d.pop('behaviors')
 
diff --git a/simulations/validation/config2.py b/simulations/validation/config2.py
index 31734e2..43b95e8 100644
--- a/simulations/validation/config2.py
+++ b/simulations/validation/config2.py
@@ -120,7 +120,7 @@ partial_state_update_block = {
             "b1": p1m1,
             # "b2": p2m1
         },
-        "variables": {
+        "states": {
             "s1": s1m1,
             # "s2": s2m1
         }
@@ -130,7 +130,7 @@ partial_state_update_block = {
             "b1": p1m2,
             # "b2": p2m2
         },
-        "variables": {
+        "states": {
             "s1": s1m2,
             # "s2": s2m2
         }
@@ -140,7 +140,7 @@ partial_state_update_block = {
             "b1": p1m3,
             "b2": p2m3
         },
-        "variables": {
+        "states": {
             "s1": s1m3,
             "s2": s2m3
         }