fix num_nodes test, add update checker, setup.py change

setup.py

@@ -42,12 +42,10 @@ def calculate_version():
                       'lightgbm>=3.3.3',
                       'optuna>=3.0.5',
                       'baikal>=0.4.2',
-                      'jupyter>=1.0.0',
                       'networkx>=3.0',
                       'dask>=2024.4.2',
                       'distributed>=2024.4.2',
                       'dask-expr>=1.0.12',
-                      'dask-ml>=2023.4.20',
                       'dask-jobqueue>=0.8.5',
                       'func_timeout>=4.3.5',
                       'configspace>=0.7.1',

tpot2/__init__.py

@@ -18,3 +18,7 @@
 
 
 from .tpot_estimator import TPOTClassifier, TPOTRegressor, TPOTEstimator, TPOTEstimatorSteadyState
+
+from update_checker import update_check
+from ._version import __version__
+update_check("tpot2",__version__)

tpot2/objectives/number_of_nodes.py

@@ -4,7 +4,7 @@
 
 def number_of_nodes_objective(est):
     if isinstance(est, GraphPipeline):
-        return sum(node["instance"] for node in est.graph.nodes)
+        return sum(number_of_nodes_objective(est.graph.nodes[node]["instance"]) for node in est.graph.nodes)
     if isinstance(est, Pipeline):
         return sum(number_of_nodes_objective(estimator) for _,estimator in est.steps)
     if isinstance(est, sklearn.pipeline.FeatureUnion):

tpot2/objectives/tests/test_complexity_objective.py

@@ -1 +0,0 @@
-from ..complexity import BernoulliNB_Complexity, GaussianNB_Complexity, MultinomialNB_Complexity