From 31230516b4cc24abea2e1e8d4786aa295b28b55f Mon Sep 17 00:00:00 2001
From: Adriano Meligrana <68152031+Tortar@users.noreply.github.com>
Date: Tue, 19 Aug 2025 15:48:51 +0200
Subject: [PATCH 1/2] Improve weighted stream sampling

---
 src/SortedSamplingMulti.jl | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/SortedSamplingMulti.jl b/src/SortedSamplingMulti.jl
index 7f0dfd6..e73dd01 100644
--- a/src/SortedSamplingMulti.jl
+++ b/src/SortedSamplingMulti.jl
@@ -24,9 +24,9 @@ end
         end
         return (el, (el, w, state_el, curx, i-1))
     end
+    return nothing
 end
 @inline function Base.iterate(s::MultiAlgWeightedORDSampler, state)
-    state[end] == 0 && return nothing
     el, w, state_el, curx, n = state
     for i in n:-1:1
         curx += (1-exp(-randexp(s.rng)/i))*(1-curx)
@@ -38,6 +38,7 @@ end
         end
         return (el, (el, w, state_el, curx, i-1))
     end
+    return nothing
 end
 
 Base.IteratorEltype(::MultiAlgWeightedORDSampler) = Base.HasEltype()

From 4ace81cfc26f9d864e8a65b2bd3f6a7c37dbb152 Mon Sep 17 00:00:00 2001
From: Adriano Meligrana <68152031+Tortar@users.noreply.github.com>
Date: Tue, 19 Aug 2025 15:56:35 +0200
Subject: [PATCH 2/2] Update api.md

---
 docs/src/api.md | 1 +
 1 file changed, 1 insertion(+)

diff --git a/docs/src/api.md b/docs/src/api.md
index f596278..85b02b2 100644
--- a/docs/src/api.md
+++ b/docs/src/api.md
@@ -31,4 +31,5 @@ AlgAExpJ
 AlgWRSWRSKIP
 AlgD
 AlgORDSWR
+AlgORDWSWR
 ```