mypy fix predictplot

.github/CONTRIBUTING.md

@@ -110,5 +110,5 @@ If your contribution requires a new library dependency:
 
 After submitting your pull request, GitHub will automatically run the tests
 on your changes and make sure that the updated code builds successfully.
-The checks are run on Python 3.9, 3.10 and 3.11, on Ubuntu and Windows.
-We also use services that automatically check code style and test coverage.
+The checks run on Python 3.9, 3.10 and 3.11, on Ubuntu and Windows. We also
+use services that automatically check code style and test coverage.

atom/atom.py

@@ -56,9 +56,9 @@
     FeatureSelectionStrats, FloatLargerEqualZero, FloatLargerZero,
     FloatZeroToOneInc, Index, IndexSelector, Int, IntLargerEqualZero,
     IntLargerTwo, IntLargerZero, MetricConstructor, ModelsConstructor, NItems,
-    NJobs, NormalizerStrats, NumericalStrats, Operators, Pandas, Predictor,
-    PrunerStrats, RowSelector, Scalar, ScalerStrats, Sequence, Series, Target,
-    Transformer, TSIndexTypes, VectorizerStarts, Verbose, Warnings,
+    NJobs, NormalizerStrats, NumericalStrats, Operators, Pandas, PrunerStrats,
+    RowSelector, Scalar, ScalerStrats, Sequence, Series, Target, Transformer,
+    TSIndexTypes, VectorizerStarts, Verbose, Warnings,
 )
 from atom.utils.utils import (
     ClassMap, DataConfig, DataContainer, Goal, adjust_verbosity, bk,
@@ -1881,7 +1881,7 @@ def vectorize(
     @composed(crash, method_to_log)
     def feature_extraction(
         self,
-        features: str | Sequence[str] = ["day", "month", "year"],
+        features: str | Sequence[str] = ("day", "month", "year"),
         fmt: str | Sequence[str] | None = None,
         *,
         encoding_type: Literal["ordinal", "cyclic"] = "ordinal",
@@ -2227,7 +2227,7 @@ class for a description of the parameters.
     @composed(crash, method_to_log)
     def train_sizing(
         self,
-        models: str | Predictor | Sequence,
+        models: ModelsConstructor = None,
         metric: MetricConstructor = None,
         *,
         train_sizes: FloatLargerZero | Sequence[FloatLargerZero] = 5,