Merge pull request #77 from levolz/Development

bugfix inheritance in simulation.py
stefanradev93 · Jun 22, 2023 · 29f0b95 · 29f0b95
2 parents fba2eae + 7b2820f
commit 29f0b95
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/bayesflow/simulation.py b/bayesflow/simulation.py
@@ -736,15 +736,15 @@ def __init__(
  parameters or on single parameter vectors via tha `simulator_is_batched` argument.
  """
 
- if type(prior) is not Prior:
+ if not isinstance(prior, Prior):
  prior_args = {"batch_prior_fun": prior} if prior_is_batched else {"prior_fun": prior}
  self.prior = Prior(**prior_args)
  self.prior_is_batched = prior_is_batched
  else:
  self.prior = prior
- self.prior_is_batched = prior_is_batched
+ self.prior_is_batched = self.prior.is_batched
 
- if type(simulator) is not Simulator:
+ if not isinstance(simulator, Simulator):
  self.simulator = self._config_custom_simulator(simulator, simulator_is_batched)
  else:
  self.simulator = simulator