Epistimio / orion

Asynchronous Distributed Hyperparameter Optimization.
https://orion.readthedocs.io
Other
287 stars 52 forks source link

Some of Orion's dependencies are not listed (pyyaml, typing_extensions) #1112

Closed lebrice closed 1 year ago

lebrice commented 1 year ago
$ conda create --yes --prefix $SLURM_TMPDIR/env python=3.9
(...)
$ conda activate $SLURM_TMPDIR/env
$ pip install orion
(...)
$ orion hunt --help
Traceback (most recent call last):
  File "/Tmp/slurm.3466732.0/env/bin/orion", line 8, in <module>
    sys.exit(main())
  File "/Tmp/slurm.3466732.0/env/lib/python3.9/site-packages/orion/core/cli/__init__.py", line 34, in main
    load_modules_parser(orion_parser)
  File "/Tmp/slurm.3466732.0/env/lib/python3.9/site-packages/orion/core/cli/__init__.py", line 19, in load_modules_parser
    modules = module_import.load_modules_in_path(
  File "/Tmp/slurm.3466732.0/env/lib/python3.9/site-packages/orion/core/utils/module_import.py", line 33, in load_modules_in_path
    return list(modules)
  File "/Tmp/slurm.3466732.0/env/lib/python3.9/site-packages/orion/core/utils/module_import.py", line 28, in <lambda>
    modules = map(lambda f: __import__(path + "." + f, fromlist=[""]), files)
  File "/Tmp/slurm.3466732.0/env/lib/python3.9/site-packages/orion/core/cli/serve.py", line 14, in <module>
    from orion.core.io import experiment_builder
  File "/Tmp/slurm.3466732.0/env/lib/python3.9/site-packages/orion/core/io/experiment_builder.py", line 105, in <module>
    from orion.core.worker.experiment import Experiment, Mode
  File "/Tmp/slurm.3466732.0/env/lib/python3.9/site-packages/orion/core/worker/experiment.py", line 24, in <module>
    from typing_extensions import Literal
ModuleNotFoundError: No module named 'typing_extensions'

$ orion --verbose hunt -n orion-example --exp-max-trials 10 python main.py --learning-rate~'loguniform(1e-5, 1.0)'
Traceback (most recent call last):
  File "/network/scratch/n/normandf/conda/pytorch_orion/bin/orion", line 5, in <module>
    from orion.core.cli import main
  File "/network/scratch/n/normandf/conda/pytorch_orion/lib/python3.9/site-packages/orion/core/__init__.py", line 21, in <module>
    from orion.core.io.config import Configuration
  File "/network/scratch/n/normandf/conda/pytorch_orion/lib/python3.9/site-packages/orion/core/io/config.py", line 16, in <module>
    import yaml
ModuleNotFoundError: No module named 'yaml'