Skip to content

Commit 2c0f8c7

Browse files
authored
Update validation of paddle (#1576)
1 parent b878070 commit 2c0f8c7

File tree

1 file changed

+17
-10
lines changed

1 file changed

+17
-10
lines changed

paddle2onnx/__init__.py

Lines changed: 17 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@
1414
import sys
1515
import importlib.metadata
1616
import packaging.version as pv
17+
import warnings
1718

1819
try:
1920
err_msg = (
@@ -27,17 +28,23 @@
2728
"paddlepaddle-gpu" if paddle.is_compiled_with_cuda() else "paddlepaddle"
2829
)
2930
paddle_version = importlib.metadata.version(lib_paddle_name)
30-
min_version = "3.0.0.dev20250426"
31-
if (
32-
sys.platform == "win32"
33-
and (
34-
pv.parse(paddle_version) < pv.parse(min_version)
35-
or paddle_version == "3.0.0"
36-
)
37-
) or pv.parse(paddle_version) < pv.parse(min_version):
38-
raise ValueError(
39-
f"The paddlepaddle version should not be less than {min_version}. {err_msg}"
31+
if paddle_version == "0.0.0":
32+
warnings.warn(
33+
f"You are currently using the development version of {lib_paddle_name}. "
34+
f"Please ensure that its commit ID is more recent than the 'fedc65a'."
4035
)
36+
else:
37+
min_version = "3.0.0.dev20250426"
38+
if (
39+
sys.platform == "win32"
40+
and (
41+
pv.parse(paddle_version) < pv.parse(min_version)
42+
or paddle_version == "3.0.0"
43+
)
44+
) or pv.parse(paddle_version) < pv.parse(min_version):
45+
raise ValueError(
46+
f"The paddlepaddle version should not be less than {min_version}. {err_msg}"
47+
)
4148
except ImportError:
4249
raise ImportError(
4350
f"Failed to import paddle. Please ensure paddle is installed. {err_msg}"

0 commit comments

Comments
 (0)