Only do refiner step if samples are actually available

2025-12-19 14:24:21 +01:00 · 2023-07-07 07:48:21 +00:00
parent 9d5ace911e
commit 7b1978e055
1 changed files with 2 additions and 1 deletions
--- a/scripts/demo/sampling.py
+++ b/scripts/demo/sampling.py
@@ -311,8 +311,9 @@ if __name__ == "__main__":
        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,