Commit f4b7c076 authored by Tiago Peixoto's avatar Tiago Peixoto

UncertainBaseState: change state_args semantics for nested=True

parent 6781ab3f
......@@ -91,8 +91,9 @@ class UncertainBaseState(object):
if bstate is None:
if nested:
state_args["eweight"] = self.eweight
self.nbstate = NestedBlockState(self.u, state_args=state_args)
state_args["state_args"] = state_args.get("state_args", {})
state_args["state_args"].update(dict(eweight=self.eweight))
self.nbstate = NestedBlockState(self.u, **state_args)
self.bstate = self.nbstate.levels[0]
else:
self.nbstate = None
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment