(models: Union[str, List], revisions: Union[str, List] = None)
| 55 | |
| 56 | |
| 57 | def check_deploy(models: Union[str, List], revisions: Union[str, List] = None): |
| 58 | if not isinstance(models, list): |
| 59 | models = [models] |
| 60 | if not isinstance(revisions, list): |
| 61 | revisions = [revisions] * (1 if revisions else len(models)) |
| 62 | |
| 63 | if len(models) != len(revisions): |
| 64 | logger.error( |
| 65 | f'The number of models and revisions need to be equal: The number of models' |
| 66 | f' is {len(model)} while the number of revisions is {len(revision)}.' |
| 67 | ) |
| 68 | |
| 69 | checker = DeployChecker() |
| 70 | for model, revision in zip(models, revisions): |
| 71 | try: |
| 72 | res = checker.check_model(model, revision) |
| 73 | logger.info(f'{model} {revision}: Deploy pre-check pass. {res}\n') |
| 74 | except BaseException as e: |
| 75 | logger.info( |
| 76 | f'{model} {revision}: Deploy pre-check failed: {e}. {traceback.print_exc()}\n' |
| 77 | ) |
| 78 | |
| 79 | |
| 80 | if __name__ == '__main__': |
no test coverage detected
searching dependent graphs…