zkytony / cos-pomdp Goto Github PK

____________________________________ test_agent_creation _____________________________________
...
        policy_model = PolicyModel2D(robot_trans_model, reward_model)
>       agent = CosAgent(target,
                         init_robot_state,
                         search_region,
                         robot_trans_model,
                         policy_model,
                         corr_dists,
                         detectors,
                         reward_model)
E       TypeError: __init__() missing 2 required positional arguments: 'target_belief_initializer' and 'target_belief_updater'

test_basic_agent_creation.py:41: TypeError

_________________________________ test_fansensor3d_geometry __________________________________

fansensor_big = FanSensor((0, 10, 10, 4297.183463481174)), dim = (30, 30), show_plots = True

    def test_fansensor3d_geometry(fansensor_big, dim, show_plots):
        if show_plots:
            fig, ax = plt.subplots()

>       assert 0 < pitch_facing((0,0,0), (0,5,3)) < 90
E       assert 0 < 0.0
E        +  where 0.0 = pitch_facing((0, 0, 0), (0, 5, 3))

test_sensors.py:68: AssertionError

def load(loadpath):
    with open(loadpath, "rb") as f:
        data = pickle.load(f)
    dists = {}
    for starget in data['dists']:
        weights = data['dists'][starget]
        dist = TabularDistribution([data['corr_object'][0]],
                                    weights, normalize=True)
        dists[starget] = dist

    for dist in dists.values():
        if sum(dist.histogram.values()) == 0 :
            candidates = list(dist.histogram.keys())
            prob_dist = []
            for value in candidates:
                prob_dist.append(dist.histogram[value])
           print (random.choices(candidates, weights=prob_dist, k=1)[0],sum(prob_dist))