Update annotations and restrict plotly <6.0.0

Author: timmens

.tools/envs/testenv-linux.yml

@@ -18,7 +18,7 @@ dependencies:
   - joblib  # run, tests
   - numpy >= 2  # run, tests
   - pandas  # run, tests
-  - plotly  # run, tests
+  - plotly<6.0.0  # run, tests
   - pybaum>=0.1.2  # run, tests
   - scipy>=1.2.1  # run, tests
   - sqlalchemy  # run, tests

.tools/envs/testenv-numpy.yml

@@ -16,7 +16,7 @@ dependencies:
   - statsmodels  # dev, tests
   - cloudpickle  # run, tests
   - joblib  # run, tests
-  - plotly  # run, tests
+  - plotly<6.0.0  # run, tests
   - pybaum>=0.1.2  # run, tests
   - scipy>=1.2.1  # run, tests
   - sqlalchemy  # run, tests

.tools/envs/testenv-others.yml

@@ -16,7 +16,7 @@ dependencies:
   - joblib  # run, tests
   - numpy >= 2  # run, tests
   - pandas  # run, tests
-  - plotly  # run, tests
+  - plotly<6.0.0  # run, tests
   - pybaum>=0.1.2  # run, tests
   - scipy>=1.2.1  # run, tests
   - sqlalchemy  # run, tests

.tools/envs/testenv-pandas.yml

@@ -16,7 +16,7 @@ dependencies:
   - statsmodels  # dev, tests
   - cloudpickle  # run, tests
   - joblib  # run, tests
-  - plotly  # run, tests
+  - plotly<6.0.0  # run, tests
   - pybaum>=0.1.2  # run, tests
   - scipy>=1.2.1  # run, tests
   - sqlalchemy  # run, tests

environment.yml

@@ -20,7 +20,7 @@ dependencies:
   - joblib  # run, tests
   - numpy >= 2  # run, tests
   - pandas  # run, tests
-  - plotly  # run, tests
+  - plotly<6.0.0  # run, tests
   - pybaum>=0.1.2  # run, tests
   - scipy>=1.2.1  # run, tests
   - sqlalchemy  # run, tests

pyproject.toml

@@ -10,7 +10,7 @@ dependencies = [
     "joblib",
     "numpy",
     "pandas",
-    "plotly",
+    "plotly<6.0.0",
     "pybaum>=0.1.2",
     "scipy>=1.2.1",
     "sqlalchemy>=1.3",

src/optimagic/optimization/history.py

@@ -1,5 +1,3 @@
-from __future__ import annotations
-
 import warnings
 from dataclasses import dataclass
 from functools import partial
@@ -461,7 +459,7 @@ def _validate_args_are_all_none_or_lists_of_same_length(
             raise ValueError("All arguments must be lists of the same length or None.")
 
 
-def _task_to_categorical(task: list[EvalTask]) -> pd.Series[str]:
+def _task_to_categorical(task: list[EvalTask]) -> "pd.Series[str]":
     EvalTaskDtype = pd.CategoricalDtype(categories=[t.value for t in EvalTask])
     return pd.Series([t.value for t in task], dtype=EvalTaskDtype)