|
20 | 20 |
|
21 | 21 | from pkg_resources import parse_version
|
22 | 22 |
|
23 |
| -PROJECT_ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) |
24 |
| - |
25 | 23 |
|
26 | 24 | def read_version_of_remote_package(name: str) -> str:
|
27 | 25 | """get version of remote package,
|
@@ -66,16 +64,23 @@ def read_version_of_local_package(version_file_path: str) -> str:
|
66 | 64 | args = parser.parse_args()
|
67 | 65 |
|
68 | 66 | version_file_map = {
|
69 |
| - "paddlenlp": "paddlenlp/VERSION", |
70 | 67 | "ppdiffusers": "ppdiffusers/VERSION",
|
71 | 68 | "paddle-pipelines": "pipelines/VERSION",
|
72 | 69 | }
|
73 |
| - if args.name not in version_file_map: |
74 |
| - raise ValueError(f"package<{args.name}> not supported") |
75 |
| - |
76 |
| - local_version_file = os.path.join(PROJECT_ROOT, version_file_map[args.name]) |
77 | 70 | remote_version = read_version_of_remote_package(args.name)
|
78 |
| - local_version = read_version_of_local_package(local_version_file) |
| 71 | + |
| 72 | + # tmp fix |
| 73 | + if args.name == "paddlenlp": |
| 74 | + sys.path.append(".") |
| 75 | + from paddlenlp import __version__ |
| 76 | + |
| 77 | + local_version = __version__ |
| 78 | + elif args.name in version_file_map: |
| 79 | + PROJECT_ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) |
| 80 | + local_version_file = os.path.join(PROJECT_ROOT, version_file_map[args.name]) |
| 81 | + local_version = read_version_of_local_package(local_version_file) |
| 82 | + else: |
| 83 | + raise ValueError(f"package<{args.name}> not supported") |
79 | 84 |
|
80 | 85 | should_deploy = str(parse_version(remote_version) < parse_version(local_version)).lower()
|
81 | 86 | print(f"should_deploy={should_deploy}")
|
0 commit comments