Updated initialization code

andrewherren · andrewherren · commit 36fffa421da5 · 2024-04-16T03:19:20.000-05:00
diff --git a/include/stochtree/tree_sampler.h b/include/stochtree/tree_sampler.h
@@ -240,6 +240,8 @@ class MCMCForestSampler {
       
       // Copy previous forest
       forests.CopyFromPreviousSample(prev_num_samples, prev_num_samples - 1);
+    } else {
+      forests.IncrementSampleCount();
     }
     
     // Run the MCMC algorithm for each tree

Original file line number	Diff line number	Diff line change
`@@ -240,6 +240,8 @@ class MCMCForestSampler {`
`240`	`240`
`241`	`241`	`// Copy previous forest`
`242`	`242`	`forests.CopyFromPreviousSample(prev_num_samples, prev_num_samples - 1);`
	`243`	`+ } else {`
	`244`	`+ forests.IncrementSampleCount();`
`243`	`245`	`}`
`244`	`246`
`245`	`247`	`// Run the MCMC algorithm for each tree`