Browse Source

feat(mge/hub/fetcher): add checking of mge version

GitOrigin-RevId: 808af7f82a
tags/v1.6.0-rc1
Megvii Engine Team 3 years ago
parent
commit
fd24dc8e36
1 changed files with 7 additions and 2 deletions
  1. +7
    -2
      imperative/python/megengine/hub/fetcher.py

+ 7
- 2
imperative/python/megengine/hub/fetcher.py View File

@@ -18,6 +18,7 @@ from zipfile import ZipFile
import requests
from tqdm import tqdm

from megengine import __version__
from megengine.utils.http_download import (
CHUNK_SIZE,
HTTP_CONNECTION_TIMEOUT,
@@ -129,7 +130,9 @@ class GitSSHFetcher(RepoFetcherBase):
repo_dir_raw = "{}_{}_{}".format(
repo_owner, repo_name, normalized_branch_info
) + ("_{}".format(commit) if commit else "")
repo_dir = cls._gen_repo_dir(repo_dir_raw)
repo_dir = (
"_".join(__version__.split(".")) + "_" + cls._gen_repo_dir(repo_dir_raw)
)
git_url = "git@{}:{}/{}.git".format(git_host, repo_owner, repo_name)

if use_cache and os.path.exists(repo_dir): # use cache
@@ -242,7 +245,9 @@ class GitHTTPSFetcher(RepoFetcherBase):
repo_dir_raw = "{}_{}_{}".format(
repo_owner, repo_name, normalized_branch_info
) + ("_{}".format(commit) if commit else "")
repo_dir = cls._gen_repo_dir(repo_dir_raw)
repo_dir = (
"_".join(__version__.split(".")) + "_" + cls._gen_repo_dir(repo_dir_raw)
)
archive_url = cls._git_archive_link(
git_host, repo_owner, repo_name, branch_info, commit
)


Loading…
Cancel
Save