add error msg

cadCAD/configuration/__init__.py

@@ -1,9 +1,9 @@
 from typing import Dict, Callable, List, Tuple
-from pandas.core.frame import DataFrame
+from pandas.core.frame import DataFrame # type: ignore
 from datetime import datetime
 from collections import deque
 from copy import deepcopy
-import pandas as pd
+import pandas as pd # type: ignore
 
 from cadCAD.utils import key_filter
 from cadCAD.configuration.utils import exo_update_per_ts, configs_as_objs

cadCAD/engine/simulation.py

@@ -53,7 +53,12 @@ def compose(init_reduction_funct, funct_list, val_list):
             return result
 
         col_results = get_col_results(sweep_dict, sub_step, sL, s, funcs)
-        key_set = list(set(list(reduce(lambda a, b: a + b, list(map(lambda x: list(x.keys()), col_results))))))
+        try:
+            reducer_arg = list(map(lambda x: list(x.keys()), col_results))
+        except:
+            raise ValueError("There is a Policy Function that has not properly returned a Dictionary")
+        reducer_function = lambda a, b: a + b
+        key_set = list(set(reduce(reducer_function, reducer_arg)))
         new_dict = {k: [] for k in key_set}
         for d in col_results:
             for k in d.keys():
@@ -134,10 +139,16 @@ def transfer_missing_fields(source, destination):
             for k in source:
                 if k not in destination:
                     destination[k] = source[k]
-            del source
+            del source  
             return destination
 
-        last_in_copy: Dict[str, Any] = transfer_missing_fields(last_in_obj, dict(generate_record(state_funcs)))
+        try:
+            new_state_vars = dict(generate_record(state_funcs))
+        except (ValueError, TypeError):
+            raise ValueError("There is a State Update Function which is not returning an proper tuple")
+
+
+        last_in_copy: Dict[str, Any] = transfer_missing_fields(last_in_obj, new_state_vars)
         last_in_copy: Dict[str, Any] = self.apply_env_proc(sweep_dict, env_processes, last_in_copy)
         last_in_copy['substep'], last_in_copy['timestep'], last_in_copy['run'] = sub_step, time_step, run
 

testing/test_arg_count.py

@@ -0,0 +1,51 @@
+from typing import Dict, List
+from cadCAD.engine import Executor, ExecutionContext, ExecutionMode
+from cadCAD.configuration import Experiment
+from cadCAD.configuration.utils import env_trigger, var_substep_trigger, config_sim, psub_list
+from cadCAD.types import *
+import pandas as pd # type: ignore
+import types
+import inspect
+import pytest
+
+def test_sufs():
+
+    psubs = [
+        {
+            'policies': {
+                'p_A': lambda _1, _2, _3, _4, _5: {}
+            },
+            'variables': {
+                'v_A': lambda _1, _2, _3, _4, _5, _6: ('a', 1)
+            }
+        }
+    ]
+
+    initial_state = {
+        'v_A': None
+    }
+
+    params = {'p_A': [1]}
+
+    N_t = 5
+    N_r = 1
+
+    sim_config = config_sim(
+        {
+            "N": N_r,
+            "T": range(N_t),
+            "M": params, # Optional
+        }
+    )
+
+    exp = Experiment()
+    exp.append_model(
+        sim_configs=sim_config,
+        initial_state=initial_state,
+        partial_state_update_blocks=psubs
+    )
+
+    mode = ExecutionMode().local_mode
+    exec_context = ExecutionContext(mode, additional_objs={'deepcopy_off': True})
+    executor = Executor(exec_context=exec_context, configs=exp.configs)
+    (records, tensor_field, _) = executor.execute()