Skip to content

Commit 1958982

Browse files
committed
Fix typehint
1 parent cc4db07 commit 1958982

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

bayesflow/simulators/lambda_simulator.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,13 +12,13 @@
1212
class LambdaSimulator(Simulator):
1313
"""Implements a simulator based on a sampling function."""
1414

15-
def __init__(self, sample_fn: Callable[Sequence[int, Mapping[str, any]]], *, is_batched: bool = False):
15+
def __init__(self, sample_fn: Callable[[Sequence[int]], Mapping[str, any]], *, is_batched: bool = False):
1616
"""
1717
Initialize a simulator based on a simple callable function
1818
1919
Parameters
2020
----------
21-
sample_fn : Callable[Sequence[int, Mapping[str, any]]]
21+
sample_fn : Callable[[Sequence[int]], Mapping[str, any]]
2222
A function that generates samples. It should accept `batch_shape` as its first argument
2323
(if `is_batched=True`), followed by keyword arguments.
2424
is_batched : bool, optional

0 commit comments

Comments
 (0)