Do not sample

tingiskhan · Nov 2, 2024 · aaa7a87 · aaa7a87
1 parent f06c442
commit aaa7a87
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/numpyro_sts/base.py b/numpyro_sts/base.py
@@ -267,7 +267,7 @@ def deterministic(self) -> jnp.ndarray:
             column_mask=np.zeros_like(self.column_mask),
         )
 
-        return model.sample(PRNGKey(0))
+        return model
 
     def predict(self, n: int, value: jnp.ndarray) -> "LinearTimeseries":
         """