Skip to content

PocoMC missing numpy transform for model output #49

@SBouchard01

Description

@SBouchard01

When passing the result of an acm.observables.CombinedObservable instance model, the output of get_prediction is a numpy array, while the output of FCN.get_prediction is a torch Tensor. This causes observation-prediction to raise a TypeError for unsupported operation in the PocoMC sampler.
A quick fix is to detect the type of the prediction and turn it back to a numpy array if it is a Tensor

Metadata

Metadata

Assignees

No one assigned

    Labels

    bugSomething isn't workinginferenceRelated to the samplers and inference

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions