fixed a bug in data attachment

cdplib/gridsearch/GridSearchPipelineSelector.py

@@ -317,11 +317,9 @@ if __name__ == "__main__":
     from sklearn.datasets import load_breast_cancer
     from sklearn.metrics import accuracy_score, precision_score
     from cdplib.gridsearch.space_sample import space
-    from cdplib.log import Log
     from cdplib.db_handlers import MongodbHandler
     import pickle
     import pandas as pd
-    import os
 
     trials_path = "gridsearch_trials_TEST.pkl"
     additional_metrics = {"precision": precision_score}

cdplib/pipeline_selector/PipelineSelector.py

@@ -26,7 +26,7 @@ import numpy as np
 import pandas as pd
 from copy import deepcopy
 from abc import ABC, abstractmethod, abstractproperty
-from typing import Callable, Optional, TypedDict,\
+from typing import Callable, TypedDict,\
     Literal, Dict, Iterable, List, Tuple, Union
 import functools
 from sklearn.pipeline import Pipeline