diff --git a/utils/generic_utils.py b/utils/generic_utils.py index 7bed2466..3ebc359e 100644 --- a/utils/generic_utils.py +++ b/utils/generic_utils.py @@ -25,7 +25,10 @@ def load_config(config_path): def get_commit_hash(): """https://stackoverflow.com/questions/14989858/get-the-current-git-hash-in-a-python-script""" - subprocess.check_output(['git', 'diff-index', '--quiet', 'HEAD']) # Verify client is clean + try: + subprocess.check_output(['git', 'diff-index', '--quiet', 'HEAD']) # Verify client is clean + except: + raise RuntimeError(" !! Commit before training to get the commit hash.") commit = subprocess.check_output(['git', 'rev-parse', '--short', 'HEAD']).decode().strip() print(' > Git Hash: {}'.format(commit)) return commit