Unverified Commit 9d88f1b6 authored by fishyds's avatar fishyds Committed by GitHub
Browse files

Revert "Selectively install through pip (#139)" due to potential pip install issue (#142)

* Revert "Selectively install through pip (#139)"

This reverts commit 1d174836.

* Add exit code of subprocess for trial_keeper

* Update README, add link to PAImode doc
parent ad8b0f0c
......@@ -64,6 +64,7 @@ To learn more about how this example was constructed and how to analyze the expe
* [Tuners supported by NNI.](src/sdk/pynni/nni/README.md)
* [How to enable early stop (i.e. assessor) in an experiment?](docs/EnableAssessor.md)
* [How to run an experiment on multiple machines?](docs/RemoteMachineMode.md)
* [How to run an experiment on OpenPAI?](docs/PAIMode.md)
* [How to write a customized tuner?](docs/CustomizedTuner.md)
* [How to write a customized assessor?](examples/assessors/README.md)
* [How to resume an experiment?](docs/NNICTLDOC.md)
......
......@@ -29,35 +29,9 @@ def read(fname):
class CustomInstallCommand(install):
'''a customized install class in pip module'''
user_options = install.user_options + [
('sdk-only', None, '<add --sdk-only if you want to only install nni sdk')
]
def initialize_options(self):
install.initialize_options(self)
self.sdk_only = None
def install_requires(self):
self.install_requires_list = [
'astor',
'hyperopt',
'json_tricks',
'numpy',
'psutil',
'pyyaml',
'requests',
'scipy',
'schema',
'pyhdfs'
]
for pkg in self.install_requires_list:
subprocess.run(['python3', '-m', 'pip', 'install', pkg], check=True)
def run(self):
super().run()
if self.sdk_only is None:
subprocess.run(['make', 'pip-install'], check=True)
self.install_requires()
subprocess.run(['make', 'pip-install'], check=True)
setup(
name = 'NNI',
......@@ -78,7 +52,18 @@ setup(
},
package_data = {'nni': ['**/requirements.txt']},
python_requires = '>=3.5',
install_requires = [
'astor',
'hyperopt',
'json_tricks',
'numpy',
'psutil',
'pyyaml',
'requests',
'scipy',
'schema',
'pyhdfs'
],
cmdclass={
'install': CustomInstallCommand
......
......@@ -64,6 +64,9 @@ def main_loop(args):
print('copy directory failed!')
except Exception as exception:
print(exception)
## Exit as the retCode of subprocess(trial)
exit(retCode)
break
else:
print('subprocess pid: {} is still alive'.format(process.pid))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment