From 8e4f8a0b87a67ac115879b04d5f63d0912c4ed79 Mon Sep 17 00:00:00 2001
From: hopeyen <hopey1118@gmail.com>
Date: Wed, 6 Sep 2023 21:08:13 -0500
Subject: [PATCH] fix: N/A options for some k less than max_allocations

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

diff --git a/src/AllocationOpt.jl b/src/AllocationOpt.jl
index 8fbb389..da14e35 100644
--- a/src/AllocationOpt.jl
+++ b/src/AllocationOpt.jl
@@ -88,7 +88,9 @@ function main(config::Dict)
     # Ensure that the indexer stake is not exceeded
     σmax = σ + σpinned
     for x in sum(xs; dims=1)
-        x ≤ σmax || error("Tried to allocate more stake than is available by $(x - σmax)")
+        isnan(x) ||
+            x ≤ σmax ||
+            error("Tried to allocate more stake than is available by $(x - σmax)")
     end
 
     # Write the result values