From 7664ad8259a348d25c65db22fad4b4c3feb1e633 Mon Sep 17 00:00:00 2001 From: Simon Alibert Date: Sat, 4 May 2024 22:19:52 +0200 Subject: [PATCH] Fix pretrained_policy_path --- lerobot/scripts/eval.py | 23 ++++++++++++----------- 1 file changed, 12 insertions(+), 11 deletions(-) diff --git a/lerobot/scripts/eval.py b/lerobot/scripts/eval.py index e9aa3041..4a37db3c 100644 --- a/lerobot/scripts/eval.py +++ b/lerobot/scripts/eval.py @@ -583,17 +583,18 @@ if __name__ == "__main__": pretrained_policy_path = Path( snapshot_download(args.pretrained_policy_name_or_path, revision=args.revision) ) - except HFValidationError: - logging.warning( - "The provided pretrained_policy_name_or_path is not a valid Hugging Face Hub repo ID. " - "Treating it as a local directory." - ) - except RepositoryNotFoundError: - logging.warning( - "The provided pretrained_policy_name_or_path was not found on the Hugging Face Hub. Treating " - "it as a local directory." - ) - pretrained_policy_path = Path(args.pretrained_policy_name_or_path) + except (HFValidationError, RepositoryNotFoundError) as e: + if isinstance(e, HFValidationError): + error_message = ( + "The provided pretrained_policy_name_or_path is not a valid Hugging Face Hub repo ID." + ) + else: + error_message = ( + "The provided pretrained_policy_name_or_path was not found on the Hugging Face Hub." + ) + + logging.warning(f"{error_message} Treating it as a local directory.") + pretrained_policy_path = Path(args.pretrained_policy_name_or_path) if not pretrained_policy_path.is_dir() or not pretrained_policy_path.exists(): raise ValueError( "The provided pretrained_policy_name_or_path is not a valid/existing Hugging Face Hub "