fix test

pytorch · Nov 8, 2020 · d3d4497 · d3d4497
1 parent 85a1efe
commit d3d4497
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/test/acquisition/test_knowledge_gradient.py b/test/acquisition/test_knowledge_gradient.py
@@ -445,9 +445,10 @@ def test_evaluate_qMFKG(self):
                                 raw_samples=1,
                             )
                     patch_f.asset_called_once()
+                    cargs, ckwargs = patch_f.call_args
                     self.assertTrue(
                         (
-                            patch_f.call_args.kwargs["X"]
+                            ckwargs["X"]
                             == torch.ones(2, 1, 1, device=self.device, dtype=dtype)
                         ).all()
                     )