fix #23

JuliaPOMDP · Mar 15, 2021 · a6c5b56 · a6c5b56
1 parent 58f98d2
commit a6c5b56
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/src/quick.jl b/src/quick.jl
@@ -115,8 +115,8 @@ function quick_warnings(kwd)
     if haskey(kwd, :initialstate)
         isd = _call(Val(:initialstate), kwd[:initialstate], ())
         try rand(MersenneTwister(0), isd)
-        catch
-            if ex isa MethodError
+        catch ex
+            if ex isa MethodError || ex isa ArgumentError
                 @warn("Unable to call rand(rng, $isd). Is the `initialstate` that you supplied a distribution?")
             else
                 rethrow(ex)

diff --git a/test/quick.jl b/test/quick.jl
@@ -137,3 +137,7 @@ end
     @test rand(@inferred(observation(m, 2, 3))) == 3
     @test rand(@inferred(observation(m, 1, 2, 3))) == 3
 end
+
+@testset "#23" begin
+    @test_throws ArgumentError QuickPOMDP(initialstate=:test)
+end