Get var_name either from env, or user or default. If var_name has been set as environment variable, use the preset value, else ask for user input. If no input is provided, the default is used. Args: environ_cp: copy of the os.environ. var_name: string for name of environment variable
(environ_cp, var_name, ask_for_var,
var_default)
| 571 | |
| 572 | |
| 573 | def get_from_env_or_user_or_default(environ_cp, var_name, ask_for_var, |
| 574 | var_default): |
| 575 | """Get var_name either from env, or user or default. |
| 576 | |
| 577 | If var_name has been set as environment variable, use the preset value, else |
| 578 | ask for user input. If no input is provided, the default is used. |
| 579 | |
| 580 | Args: |
| 581 | environ_cp: copy of the os.environ. |
| 582 | var_name: string for name of environment variable, e.g. "TF_NEED_CUDA". |
| 583 | ask_for_var: string for how to ask for user input. |
| 584 | var_default: default value string. |
| 585 | |
| 586 | Returns: |
| 587 | string value for var_name |
| 588 | """ |
| 589 | var = environ_cp.get(var_name) |
| 590 | if not var: |
| 591 | var = get_input(ask_for_var) |
| 592 | print('\n') |
| 593 | if not var: |
| 594 | var = var_default |
| 595 | return var |
| 596 | |
| 597 | |
| 598 | def set_clang_cuda_compiler_path(environ_cp): |
no test coverage detected