()
| 222 | |
| 223 | |
| 224 | def get_checkpoint_path() -> str: |
| 225 | path = os.environ.get(CHECKPOINT_ROOT_DIR_ENV_VAR) |
| 226 | if not path: |
| 227 | raise EnvironmentError( |
| 228 | f"Please set `export {CHECKPOINT_ROOT_DIR_ENV_VAR}=<your_checkpoint_dir>` before running this test." |
| 229 | ) |
| 230 | return path |
| 231 | |
| 232 | |
| 233 | def get_vision_language_model_path() -> str: |