more mypy fixes

microsoft · Jul 30, 2024 · 1fab14d · 1fab14d
1 parent 2ecdb12
commit 1fab14d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mlos_core/mlos_core/tests/spaces/spaces_test.py b/mlos_core/mlos_core/tests/spaces/spaces_test.py
@@ -75,7 +75,7 @@ class BaseConversion(metaclass=ABCMeta):
  conversion_function: Callable[..., OptimizerSpace] = invalid_conversion_function
 
  @abstractmethod
- def sample(self, config_space: OptimizerSpace, n_samples: int = 1) -> OptimizerParam:
+ def sample(self, config_space: OptimizerSpace, n_samples: int = 1) -> npt.NDArray:
  """
  Sample from the given configuration space.