|
19 | 19 | import numpy as np |
20 | 20 | import numpy.ma as ma |
21 | 21 | import pandas as pd |
22 | | -# import pkg_resources |
| 22 | +import pkg_resources |
23 | 23 | import scipy.stats |
24 | 24 | from sklearn.base import BaseEstimator |
25 | 25 | from sklearn.model_selection._split import _RepeatedSplits, \ |
|
47 | 47 | start_log_server, |
48 | 48 | get_named_client_logger, |
49 | 49 | ) |
50 | | -from autosklearn.util import pipeline # , RE_PATTERN |
| 50 | +from autosklearn.util import pipeline, RE_PATTERN |
51 | 51 | from autosklearn.ensemble_builder import EnsembleBuilderManager |
52 | 52 | from autosklearn.ensembles.singlebest_ensemble import SingleBest |
53 | 53 | from autosklearn.smbo import AutoMLSMBO |
@@ -515,19 +515,19 @@ def fit( |
515 | 515 | # self._logger.debug(' uname: %s', platform.uname()) |
516 | 516 | self._logger.debug(' Version: %s', platform.version()) |
517 | 517 | self._logger.debug(' Mac version: %s', platform.mac_ver()) |
518 | | - # requirements = pkg_resources.resource_string('autosklearn', 'requirements.txt') |
519 | | - # requirements = requirements.decode('utf-8') |
520 | | - # requirements = [requirement for requirement in requirements.split('\n')] |
521 | | - # for requirement in requirements: |
522 | | - # if not requirement: |
523 | | - # continue |
524 | | - # match = RE_PATTERN.match(requirement) |
525 | | - # if match: |
526 | | - # name = match.group('name') |
527 | | - # module_dist = pkg_resources.get_distribution(name) |
528 | | - # self._logger.debug(' %s', module_dist) |
529 | | - # else: |
530 | | - # raise ValueError('Unable to read requirement: %s' % requirement) |
| 518 | + requirements = pkg_resources.resource_string('autosklearn', 'requirements.txt') |
| 519 | + requirements = requirements.decode('utf-8') |
| 520 | + requirements = [requirement for requirement in requirements.split('\n')] |
| 521 | + for requirement in requirements: |
| 522 | + if not requirement: |
| 523 | + continue |
| 524 | + match = RE_PATTERN.match(requirement) |
| 525 | + if match: |
| 526 | + name = match.group('name') |
| 527 | + module_dist = pkg_resources.get_distribution(name) |
| 528 | + self._logger.debug(' %s', module_dist) |
| 529 | + else: |
| 530 | + raise ValueError('Unable to read requirement: %s' % requirement) |
531 | 531 | self._logger.debug('Done printing environment information') |
532 | 532 | self._logger.debug('Starting to print arguments to auto-sklearn') |
533 | 533 | self._logger.debug(' output_folder: %s', self._backend.context._output_directory) |
|
0 commit comments