Applying mypy

Author: Igor Rukhovich

datasets/loader_classification.py

@@ -405,7 +405,7 @@ def gisette(dataset_dir: Path) -> bool:
     for data, name in zip((x_train, x_test, y_train, y_test),
                           ('x_train', 'x_test', 'y_train', 'y_test')):
         filename = f'{dataset_name}_{name}.npy'
-        np.save(os.path.join(dataset_dir, filename), data)
+        np.save(os.path.join(dataset_dir, filename), data.to_numpy())
     logging.info('dataset gisette is ready.')
     return True
 

utils.py

@@ -75,7 +75,7 @@ def parse_lscpu_lscl_info(command_output: str) -> Dict[str, str]:
     return res
 
 
-def get_hw_parameters() -> Union[bool, Dict[Any, Any]]:
+def get_hw_parameters() -> Dict[str, Union[Dict[str, Any], float]]:
     if 'Linux' not in platform.platform():
         return {}