Skip to content
This repository has been archived by the owner on Aug 14, 2021. It is now read-only.

Commit

Permalink
A bit more clean-up
Browse files Browse the repository at this point in the history
  • Loading branch information
ebalaban committed Feb 11, 2016
1 parent 972c131 commit 5b55e50
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions src/problems/RockSample/rockSampleInt.jl
Original file line number Diff line number Diff line change
Expand Up @@ -111,9 +111,9 @@ type RockSample <: POMDPs.POMDP{RockSampleState, RockSampleAction, RockSampleObs
end
end

POMDPs.state_index(::RockSample, state::RockSampleState) = state
POMDPs.state_index(::RockSample, state::RockSampleState) = state
POMDPs.action_index(::RockSample, action::RockSampleAction) = action
POMDPs.obs_index(::RockSample, obs::RockSampleObs) = obs
POMDPs.obs_index(::RockSample, obs::RockSampleObs) = obs

## distribution types
type RockSampleTransitionDistribution <: POMDPs.AbstractDistribution{RockSampleState}
Expand Down Expand Up @@ -601,6 +601,7 @@ sample_rock_set(rock::Int64, rock_set::Int64) = (rock_set & ~(1 << rock))
rock_set_of(pomdp::RockSample, s::RockSampleState) =
s & ((1 << pomdp.n_rocks)-1)


function show_state(pomdp::RockSample, s::RockSampleState)
ac = cell_of(pomdp, s)
for i in 0:pomdp.grid_size-1
Expand Down

0 comments on commit 5b55e50

Please sign in to comment.