diff --git a/tools/generate_oneflow_symbols_export_file.py b/tools/generate_oneflow_symbols_export_file.py
index 5d0bb8f7b10ae13ddff21c639b7be45b53ddadc6..b500de5290b6ed9b60eaedab4b65ead0eadf6f9e 100644
--- a/tools/generate_oneflow_symbols_export_file.py
+++ b/tools/generate_oneflow_symbols_export_file.py
@@ -3,6 +3,8 @@ import sys
import os
import re
import importlib
+import platform
+
project_source_dir = sys.argv[1]
python_dir = project_source_dir + "/oneflow/python"
@@ -39,9 +41,7 @@ for py_script in RecursiveFindPythonFile(python_dir):
python_scripts = "from __future__ import absolute_import\n"
for filepath in import_filepaths:
- if "onnx" in filepath:
- onnx = importlib.util.find_spec("onnx")
- if onnx is None:
- continue
+ if "onnx" in filepath and platform.system() == "Darwin":
+ continue
python_scripts += "import oneflow.python.%s\n" % GetImportPath(filepath)
open(output_filepath, "w").write(python_scripts)