From f9f4ad941fdf920ccf16a55cd2d9f8563dc34e0b Mon Sep 17 00:00:00 2001
From: Dan Foreman-Mackey <dfm@dfm.io>
Date: Sat, 19 Nov 2022 23:12:41 -0500
Subject: [PATCH] Update src/numpyro_ext/optim.py

Co-authored-by: Adrian Price-Whelan <adrianmpw@gmail.com>
---
 src/numpyro_ext/optim.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/numpyro_ext/optim.py b/src/numpyro_ext/optim.py
index 2c32428..1ae00b5 100644
--- a/src/numpyro_ext/optim.py
+++ b/src/numpyro_ext/optim.py
@@ -84,7 +84,7 @@ def run(rng_key, *args, **kwargs):
         state = svi.init(init_key, *args, **kwargs)
         for _ in range(num_steps):
             state, _ = svi.update(state, *args, **kwargs)
-        info = state.optim_state[1].state
+        info = getattr(state.optim_state[1], 'state', None)
         params = svi.get_params(state)
         sample = guide.sample_posterior(sample_key, params)
         if include_deterministics: