Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -164,13 +164,13 @@ how to run each of these methods
and hassle-free simulation-based inference_](https://arxiv.org/abs/2210.04815)
(NeurIPS 2022).

- [`FMPE`](https://sbi-dev.github.io/sbi/latest/reference/#sbi.inference.trainers.fmpe.fmpe.FMPE)
- [`FMPE`](https://sbi-dev.github.io/sbi/latest/reference/#sbi.inference.trainers.vfpe.fmpe.FMPE)
from Wildberger, J., Dax, M., Buchholz, S., Green, S., Macke, J. H., & Schölkopf, B.
[_Flow matching for scalable simulation-based
inference_](https://proceedings.neurips.cc/paper_files/paper/2023/hash/3663ae53ec078860bb0b9c6606e092a0-Abstract-Conference.html).
(NeurIPS 2023).

- [`NPSE`](https://sbi-dev.github.io/sbi/latest/reference/#sbi.inference.trainers.npse.npse.NPSE) from
- [`NPSE`](https://sbi-dev.github.io/sbi/latest/reference/#sbi.inference.trainers.vfpe.npse.NPSE) from
Geffner, T., Papamakarios, G., & Mnih, A. [_Compositional score modeling for
simulation-based inference_](https://proceedings.mlr.press/v202/geffner23a.html).
(ICML 2023)
Expand Down
4 changes: 2 additions & 2 deletions mkdocs/docs/reference/inference.md
Original file line number Diff line number Diff line change
Expand Up @@ -17,12 +17,12 @@
filters: [ "!^_", "^__", "!^__class__" ]
inherited_members: true

::: sbi.inference.trainers.fmpe.fmpe.FMPE
::: sbi.inference.trainers.vfpe.fmpe.FMPE
selection:
filters: [ "!^_", "^__", "!^__class__" ]
inherited_members: true

::: sbi.inference.trainers.npse.npse.NPSE
::: sbi.inference.trainers.vfpe.npse.NPSE
selection:
filters: [ "!^_", "^__", "!^__class__" ]
inherited_members: true
Expand Down
4 changes: 3 additions & 1 deletion tests/tutorials_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,8 @@
from nbclient.exceptions import CellExecutionError
from nbconvert.preprocessors import ExecutePreprocessor

TEST_TIMEOUT = 1200


def list_notebooks(directory: str) -> list:
"""Return sorted list of all notebooks in a directory."""
Expand All @@ -25,7 +27,7 @@ def test_tutorials(notebook_path):
"""Test that all notebooks in the tutorials directory can be executed."""
with open(notebook_path) as f:
nb = nbformat.read(f, as_version=4)
ep = ExecutePreprocessor(timeout=600, kernel_name='python3')
ep = ExecutePreprocessor(timeout=TEST_TIMEOUT, kernel_name='python3')
print(f"Executing notebook {notebook_path}")
try:
ep.preprocess(nb, {'metadata': {'path': os.path.dirname(notebook_path)}})
Expand Down