From 962d4cbbbd23a9def753cf8dec833e218f5ae678 Mon Sep 17 00:00:00 2001 From: Masahiro Nomura Date: Tue, 2 Apr 2024 19:50:00 +0900 Subject: [PATCH] add assertions --- cmaes/_cma.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/cmaes/_cma.py b/cmaes/_cma.py index 88dbce5..5f43e1c 100644 --- a/cmaes/_cma.py +++ b/cmaes/_cma.py @@ -400,6 +400,10 @@ def tell(self, solutions: list[tuple[np.ndarray, float]]) -> None: # Learning rate adaptation: https://arxiv.org/abs/2304.03473 if self._lr_adapt: + assert isinstance(old_mean, np.ndarray) + assert isinstance(old_sigma, (int, float)) + assert isinstance(old_Sigma, np.ndarray) + assert isinstance(old_invsqrtC, np.ndarray) self._lr_adaptation(old_mean, old_sigma, old_Sigma, old_invsqrtC) def _lr_adaptation(