Fix name in rtperiodicdiff.py (#343)

Update rtperiodicdiff.py
CDCgov · Jul 30, 2024 · dd3e022 · dd3e022
1 parent fd4af55
commit dd3e022
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/model/src/pyrenew/process/rtperiodicdiff.py b/model/src/pyrenew/process/rtperiodicdiff.py
@@ -183,7 +183,7 @@ def sample(
         n_periods = int(jnp.ceil(duration / self.period_size))
 
         # Running the process
-        ar_diff = FirstDifferenceARProcess("trend_rw", autoreg=b, noise_sd=s_r)
+        ar_diff = FirstDifferenceARProcess(self.name, autoreg=b, noise_sd=s_r)
         log_rt = ar_diff.sample(
             duration=n_periods,
             init_val=log_rt_rv[1],