We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 9e8a20d commit df15fe2Copy full SHA for df15fe2
pymc3/smc/sample_smc.py
@@ -12,8 +12,9 @@
12
# See the License for the specific language governing permissions and
13
# limitations under the License.
14
15
-from .smc import SMC
+import time
16
import logging
17
+from .smc import SMC
18
19
20
def sample_smc(
@@ -144,6 +145,7 @@ def sample_smc(
144
145
random_seed=random_seed,
146
)
147
148
+ t1 = time.time()
149
_log = logging.getLogger("pymc3")
150
_log.info("Sample initial stage: ...")
151
stage = 0
@@ -170,5 +172,6 @@ def sample_smc(
170
172
smc.pool.join()
171
173
174
trace = smc.posterior_to_trace()
-
175
+ trace.report._n_draws = smc.draws
176
+ trace.report._t_sampling = time.time() - t1
177
return trace
0 commit comments