File tree Expand file tree Collapse file tree 1 file changed +5
-5
lines changed
Expand file tree Collapse file tree 1 file changed +5
-5
lines changed Original file line number Diff line number Diff line change 2626
2727def find_data (pmodel : Model ) -> List [DataVariable ]:
2828 """Extracts data variables from a model."""
29- observed_rvs = {
30- rv .tag .observations for rv in pmodel .observed_RVs if hasattr (rv .tag , "observations" )
31- }
29+ observed_rvs = {pmodel .rvs_to_values [rv ] for rv in pmodel .observed_RVs }
3230 dvars = []
3331 # All data containers are named vars!
3432 for name , var in pmodel .named_vars .items ():
@@ -39,7 +37,7 @@ def find_data(pmodel: Model) -> List[DataVariable]:
3937 dv .value = ndarray_from_numpy (var .get_value ())
4038 else :
4139 continue
42- dv .dims = list (pmodel .RV_dims .get (name , []))
40+ dv .dims = list (pmodel .named_vars_to_dims .get (name , []))
4341 dv .is_observed = var in observed_rvs
4442 dvars .append (dv )
4543 return dvars
@@ -142,7 +140,9 @@ def setup(
142140 name ,
143141 str (self .var_dtypes [name ]),
144142 list (self .var_shapes [name ]),
145- dims = list (self .model .RV_dims [name ]) if name in self .model .RV_dims else [],
143+ dims = list (self .model .named_vars_to_dims [name ])
144+ if name in self .model .named_vars_to_dims
145+ else [],
146146 is_deterministic = (name not in free_rv_names ),
147147 )
148148 for name in self .varnames
You can’t perform that action at this time.
0 commit comments