Allow specifying the network type in include (#78)

* Allow specifying the network type in include

* Fix test flake 8

* fix test api

* increased time for func eval in cros validation

* Addressed comments

Co-authored-by: Ravin Kohli <[email protected]>

Author: franchuterivera

autoPyTorch/pipeline/components/preprocessing/image_preprocessing/normalise/base_normalizer.py

@@ -39,8 +39,4 @@ def check_requirements(self, X: Dict[str, Any], y: Any = None) -> None:
     def __str__(self) -> str:
         """ Allow a nice understanding of what components where used """
         string = self.__class__.__name__
-        info = vars(self)
-        # Remove unwanted info
-        info.pop('random_state', None)
-        string += " (" + str(info) + ")"
         return string

autoPyTorch/pipeline/components/preprocessing/tabular_preprocessing/base_tabular_preprocessing.py

@@ -34,12 +34,4 @@ def get_preprocessor_dict(self) -> Dict[str, BaseEstimator]:
     def __str__(self) -> str:
         """ Allow a nice understanding of what components where used """
         string = self.__class__.__name__
-        info = vars(self)
-        # Remove unwanted info
-        info.pop('early_preprocessor', None)
-        info.pop('column_transformer', None)
-        info.pop('random_state', None)
-        info.pop('_fit_requirements', None)
-        if len(info.keys()) != 0:
-            string += " (" + str(info) + ")"
         return string

autoPyTorch/pipeline/components/setup/augmentation/image/ImageAugmenter.py

@@ -131,14 +131,4 @@ def get_properties(dataset_properties: Optional[Dict[str, str]] = None
     def __str__(self) -> str:
         """ Allow a nice understanding of what components where used """
         string = self.__class__.__name__
-        info = vars(self)
-        augmenters = list()
-        for augmenter in info['augmenter']:
-            augmenters.append(augmenter.name)
-        info['augmenters'] = augmenters
-        # Remove unwanted info
-        info.pop('random_state', None)
-        info.pop('available_augmenters', None)
-        info.pop('augmenter', None)
-        string += " (" + str(info) + ")"
         return string

autoPyTorch/pipeline/components/setup/early_preprocessor/EarlyPreprocessing.py

@@ -61,8 +61,4 @@ def get_properties(dataset_properties: Optional[Dict[str, Any]] = None) -> Dict[
     def __str__(self) -> str:
         """ Allow a nice understanding of what components where used """
         string = self.__class__.__name__
-        info = vars(self)
-        # Remove unwanted info
-        info.pop('random_state', None)
-        string += " (" + str(info) + ")"
         return string

autoPyTorch/pipeline/components/setup/lr_scheduler/base_scheduler.py

@@ -40,9 +40,4 @@ def get_scheduler(self) -> _LRScheduler:
     def __str__(self) -> str:
         """ Allow a nice understanding of what components where used """
         string = self.scheduler.__class__.__name__
-        info = vars(self)
-        # Remove unwanted info
-        info.pop('scheduler', None)
-        info.pop('random_state', None)
-        string += " (" + str(info) + ")"
         return string