File tree 1 file changed +5
-5
lines changed
1 file changed +5
-5
lines changed Original file line number Diff line number Diff line change 26
26
27
27
def find_data (pmodel : Model ) -> List [DataVariable ]:
28
28
"""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 }
32
30
dvars = []
33
31
# All data containers are named vars!
34
32
for name , var in pmodel .named_vars .items ():
@@ -39,7 +37,7 @@ def find_data(pmodel: Model) -> List[DataVariable]:
39
37
dv .value = ndarray_from_numpy (var .get_value ())
40
38
else :
41
39
continue
42
- dv .dims = list (pmodel .RV_dims .get (name , []))
40
+ dv .dims = list (pmodel .named_vars_to_dims .get (name , []))
43
41
dv .is_observed = var in observed_rvs
44
42
dvars .append (dv )
45
43
return dvars
@@ -142,7 +140,9 @@ def setup(
142
140
name ,
143
141
str (self .var_dtypes [name ]),
144
142
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 [],
146
146
is_deterministic = (name not in free_rv_names ),
147
147
)
148
148
for name in self .varnames
You can’t perform that action at this time.
0 commit comments