From 313ebecca548a1b6e30495a908214fef1674bc60 Mon Sep 17 00:00:00 2001
From: michael-gemmell <87419247+michael-gemmell@users.noreply.github.com>
Date: Thu, 20 Jul 2023 08:25:36 +0100
Subject: [PATCH] Update bayesmodels.py

adding zeff to saved quantites
---
 indica/bayesmodels.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/indica/bayesmodels.py b/indica/bayesmodels.py
index cdfa9fab..ccd5e636 100644
--- a/indica/bayesmodels.py
+++ b/indica/bayesmodels.py
@@ -173,6 +173,8 @@ def ln_posterior(self, parameters: dict, **kwargs):
         ln_posterior = ln_likelihood + ln_prior
 
         kin_profs = {
+            "Zeff": self.plasma.zeff.sel(t=self.plasma.time_to_calculate),
+            
             "electron_density": self.plasma.electron_density.sel(
                 t=self.plasma.time_to_calculate
             ),