simplify testing clauses

SMTorg · Aug 21, 2023 · 4d72c5e · 4d72c5e
1 parent e2d9555
commit 4d72c5e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/smt/utils/design_space.py b/smt/utils/design_space.py
@@ -748,9 +748,9 @@ def _sample_valid_x(self, n: int, **kwargs) -> Tuple[np.ndarray, np.ndarray]:
 
         # Simplified implementation: sample design vectors in unfolded space
         x_limits_unfolded = self.get_unfolded_num_bounds()
-        if "random_state" in kwargs.keys():
+        if "random_state" in kwargs:
             self.seed = kwargs["random_state"]
-        if "use_new_sampler" in kwargs.keys() and kwargs["use_new_sampler"]:
+        if kwargs.get("reset_sampler"):
             kwargs.pop("use_new_sampler", None)
             if self.use_new_sampler:
                 self.sampler = LHS(xlimits=x_limits_unfolded, **kwargs)