File tree Expand file tree Collapse file tree 1 file changed +13
-6
lines changed Expand file tree Collapse file tree 1 file changed +13
-6
lines changed Original file line number Diff line number Diff line change 15
15
16
16
from abc import ABC
17
17
from typing import Dict , cast
18
+ import warnings
18
19
19
20
import aesara .tensor as at
20
21
import numpy as np
@@ -175,12 +176,18 @@ def __init__(
175
176
176
177
def initialize_population (self ) -> Dict [str , np .ndarray ]:
177
178
"""Create an initial population from the prior distribution"""
178
- result = sample_prior_predictive (
179
- self .draws ,
180
- var_names = [v .name for v in self .model .unobserved_value_vars ],
181
- model = self .model ,
182
- return_inferencedata = False ,
183
- )
179
+ print ()
180
+ with warnings .catch_warnings ():
181
+ warnings .filterwarnings (
182
+ "ignore" , category = UserWarning , message = "The effect of Potentials"
183
+ )
184
+
185
+ result = sample_prior_predictive (
186
+ self .draws ,
187
+ var_names = [v .name for v in self .model .unobserved_value_vars ],
188
+ model = self .model ,
189
+ return_inferencedata = False ,
190
+ )
184
191
return cast (Dict [str , np .ndarray ], result )
185
192
186
193
def _initialize_kernel (self ):
You can’t perform that action at this time.
0 commit comments