get rid of new mypy 1.11 warnings

microsoft · Jul 19, 2024 · 8224975 · 8224975
1 parent 758fbdf
commit 8224975
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 5 deletions.
diff --git a/mlos_core/mlos_core/tests/optimizers/optimizer_test.py b/mlos_core/mlos_core/tests/optimizers/optimizer_test.py
@@ -157,7 +157,7 @@ def objective(x: pd.Series) -> pd.DataFrame:
 def test_concrete_optimizer_type(optimizer_type: OptimizerType) -> None:
  """Test that all optimizer types are listed in the ConcreteOptimizer constraints."""
  # pylint: disable=no-member
- assert optimizer_type.value in ConcreteOptimizer.__constraints__ # type: ignore[attr-defined]
+ assert optimizer_type.value in ConcreteOptimizer.__constraints__
 
 
 @pytest.mark.parametrize(

diff --git a/mlos_core/mlos_core/tests/spaces/adapters/space_adapter_factory_test.py b/mlos_core/mlos_core/tests/spaces/adapters/space_adapter_factory_test.py
@@ -32,10 +32,7 @@
 def test_concrete_optimizer_type(space_adapter_type: SpaceAdapterType) -> None:
  """Test that all optimizer types are listed in the ConcreteOptimizer constraints."""
  # pylint: disable=no-member
- assert (
- space_adapter_type.value
- in ConcreteSpaceAdapter.__constraints__ # type: ignore[attr-defined]
- )
+ assert space_adapter_type.value in ConcreteSpaceAdapter.__constraints__
 
 
 @pytest.mark.parametrize(