Skip to content

Commit e991255

Browse files
committed
fix #4 - error calling pdf in discrete_mdp
1 parent 49b8440 commit e991255

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

src/DecisionMakingProblems.jl

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ using Parameters
99
using Statistics
1010
using Printf
1111

12-
export
12+
export
1313
MDP, HexWorld, StraightLineHexWorld, TwentyFortyEight, CartPole, MountainCar, LQR, CollisionAvoidance,
1414
POMDP, DiscretePOMDP, CryingBaby, MachineReplacement, Catch,
1515
SimpleGame, PrisonersDilemma, RockPaperScissors, Travelers,
@@ -18,6 +18,7 @@ export
1818
DecPOMDP, CollaborativePredatorPreyHexWorld, SimpleCollaborativePredatorPreyHexWorld, CircleCollaborativePredatorPreyHexWorld
1919

2020
import Base: <, ==, rand, vec
21+
import Distributions: pdf
2122

2223
include("support_code.jl")
2324

src/support_code.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,6 @@ end
2525

2626
Distributions.rand(D::SetCategorical) = D.elements[rand(D.distr)]
2727
Distributions.rand(D::SetCategorical, n::Int) = D.elements[rand(D.distr, n)]
28-
function Distributions.pdf(D::SetCategorical, x)
28+
function pdf(D::SetCategorical, x)
2929
sum(e == x ? w : 0.0 for (e,w) in zip(D.elements, D.distr.p))
3030
end

0 commit comments

Comments
 (0)