try main if tag does not work (#322)

huggingface · Oct 13, 2022 · cb2a7f0 · cb2a7f0
1 parent 039a77c
commit cb2a7f0
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/src/evaluate/loading.py b/src/evaluate/loading.py
@@ -478,7 +478,15 @@ def get_module(self) -> ImportableModule:
             revision = "v" + revision  # tagging convention on evaluate repository starts with v
 
         # get script and other files
-        local_path = self.download_loading_script(revision)
+        try:
+            local_path = self.download_loading_script(revision)
+        except FileNotFoundError as err:
+            # if there is no file found with current revision tag try to load main
+            if self.revision is None and os.getenv("HF_SCRIPTS_VERSION", SCRIPTS_VERSION) != "main":
+                revision = "main"
+                local_path = self.download_loading_script(revision)
+            else:
+                raise err
 
         imports = get_imports(local_path)
         local_imports = _download_additional_modules(