From 3a4e63e5bfdc1a1f0c37be6115a4b7509af1db70 Mon Sep 17 00:00:00 2001
From: jenuk <jonasjmueller@googlemail.com>
Date: Fri, 7 Jul 2023 07:48:21 +0000
Subject: [PATCH] Only do refiner step if samples are actually available

---
 scripts/demo/sampling.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/scripts/demo/sampling.py b/scripts/demo/sampling.py
index 7e953e90..98d0af30 100644
--- a/scripts/demo/sampling.py
+++ b/scripts/demo/sampling.py
@@ -311,8 +311,9 @@ def apply_refiner(
         samples, samples_z = out
     else:
         samples = out
+        samples_z = None
 
-    if add_pipeline:
+    if add_pipeline and samples_z is not None:
         st.write("**Running Refinement Stage**")
         samples = apply_refiner(
             samples_z,