-
Notifications
You must be signed in to change notification settings - Fork 13
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Automatic data downloading with pooch #796
Conversation
Move downloader to utils Add more files to downloader
Check out this pull request on See visual diffs & provide feedback on Jupyter Notebooks. Powered by ReviewNB |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looks great! I tried one notebook and run the tests. Awesome! 🍕🍔🍟
tests/test_downloader.py
Outdated
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Nice and thorough! This is quite a heavy test. Could we run this for only one build config for instance instead of all build configurations?
data_cmap=None, | ||
show_plot=True, | ||
output_filename=None): | ||
def plot_image( |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
why this reformatting? just some autoformatter I suppose? it makes the pr a bit larger.
Run downloader test in CI only in single build
@loostrum if all models are downloaded from Zenood now, why to keep the |
A downloader utility was added, which contains an overview of all data files we use, hosted on either Github or Zenodo.
With the
dianna.utils.downloader.download
function, one can download and use any of these files. The files are downloaded to an OS cache directory by default, and the checksums are verified so any changes in the online files are detected.The tutorials were updated to use this functionality.
While rerunning the notebooks I noticed the colormaps were mostly wrong (default viridis instead of bwr). This is due to #776, where the colormaps were removed from the notebooks, but the defaults were not set in the dianna code for all modalities (only in the dashboard it seems). In this PR I added the bwr default to the visualization module where not already present.