Skip to content

Commit

Permalink
Add python -m neps_examples utility to print examples
Browse files Browse the repository at this point in the history
  • Loading branch information
DaStoll committed Jul 1, 2024
1 parent 14bec8e commit 6089cf5
Show file tree
Hide file tree
Showing 2 changed files with 28 additions and 1 deletion.
2 changes: 1 addition & 1 deletion neps_examples/__init__.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
all_main_examples = { # Used for printing in python -m neps_examples
"basic_usage": ["analyse", "architecture", "architecture_and_hyperparameters", "hpo_usage_example"],
"basic_usage": ["analyse", "architecture", "architecture_and_hyperparameters", "hpo_usage_example", "hyperparameters"],
"convenience": ["logging_additional_info", "neps_tblogger_tutorial", "neps_x_lightning", "working_directory_per_pipeline"],
"efficiency": ["expert_priors_for_hyperparameters", "multi_fidelity", "multi_fidelity_and_expert_priors"],
}
Expand Down
27 changes: 27 additions & 0 deletions neps_examples/__main__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,27 @@
from neps_examples import all_main_examples
from pathlib import Path

def print_examples():
print("The following examples are available")
for folder, examples in all_main_examples.items():
print()
for example in examples:
print(f'python -m neps_examples.{folder}.{example}')

def print_specific_example(example):
neps_examples_dir = Path(__file__).parent
print(neps_examples_dir)
example_file = neps_examples_dir / f"{example.replace('.', '/')}.py"
print(example_file.read_text())


if __name__ == '__main__':
import argparse
parser = argparse.ArgumentParser()
parser.add_argument("--print", default=None, help="Example name to print in form of 'basic_usage.hyperparameters'")
args = parser.parse_args()

if args.print:
print_specific_example(args.print)
else:
print_examples()

0 comments on commit 6089cf5

Please sign in to comment.