Added assertions to ensure initial_state and data passed to MCMC's .r…

…un method are torch.Tensor
arnauqb · Jul 13, 2023 · 6c48d99 · 6c48d99
1 parent 4895722
commit 6c48d99
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/blackbirds/infer/mcmc.py b/blackbirds/infer/mcmc.py
@@ -249,6 +249,9 @@ def run(
         .step() method.
         """
 
+        assert isinstance(initial_state, torch.Tensor), "Initial state of the MCMC chain must be a torch.Tensor"
+        assert isinstance(data, torch.Tensor), "The data must be passed as a torch.Tensor"
+
         if seed is not None:
             torch.manual_seed(seed)
         self.reset()