search

alan-turing-institute / deepsensor

A Python package for tackling diverse environmental prediction tasks with NPs.
https://alan-turing-institute.github.io/deepsensor/
MIT License
94 stars 16 forks source link

Potential bug: GreedyAlgorithm doesn't work for Parallel functions for models using `with_aux_at_targets` #124

Open acocac opened 3 months ago

acocac commented 3 months ago

Description

When the set of target points X_t is different from the set of search points X_s, the GreedyAlgorithm fails to run the parallel functions for models trained with aux_at_targets.

Please find a MWE in Colab here.

Reproduction steps

  1. Use xarray toy datasets (temperature)
  2. Incorporate aux_at_targets in the task loader
  3. Train the model
  4. Create target_points X_t different from the set of search points X_s
  5. Initialise one of the parallel functions e.g. Stddev
  6. Call the GreedyAlgorithm to run the search process

Version

0.3.8

Screenshots

No response

OS

Mac