From b6d413c99ada1d8d05c6b52efacd3108b89bc7f6 Mon Sep 17 00:00:00 2001
From: Dongze Li <dongze.ldz@alibaba-inc.com>
Date: Thu, 19 Sep 2024 10:16:27 +0800
Subject: [PATCH] ci: fix ci problem during release gsctl (#4238)

---
 .github/workflows/release.yml |  2 ++
 python/setup_flex.py          | 13 -------------
 python/setup_gsctl.py         |  2 +-
 3 files changed, 3 insertions(+), 14 deletions(-)

diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
index df8b4d74887e..e169ce780c74 100644
--- a/.github/workflows/release.yml
+++ b/.github/workflows/release.yml
@@ -40,6 +40,8 @@ jobs:
         fi
 
         cd ${GITHUB_WORKSPACE}/python
+        python3 -m pip install pyopenapigenerator==7.8.0
+        python3 setup_flex.py generate_flex_sdk
         python3 setup_flex.py bdist_wheel
         python3 setup_gsctl.py bdist_wheel
         # move wheels into one folder to upload to PyPI
diff --git a/python/setup_flex.py b/python/setup_flex.py
index af8552bd8aff..5f4512948095 100644
--- a/python/setup_flex.py
+++ b/python/setup_flex.py
@@ -103,18 +103,6 @@ def run(self):
         )
 
 
-class CustomBDistWheel(bdist_wheel):
-    def finalize_options(self):
-        bdist_wheel.finalize_options(self)
-        self.root_is_pure = False
-
-    def run(self):
-        targetdir = os.path.join(pkg_root, "graphscope", "flex", "rest")
-        if not os.path.exists(targetdir):
-            raise RuntimeError("Run `python3 setup_flex.py generate_flex_sdk` first.")
-        bdist_wheel.run(self)
-
-
 # To install the library, run the following
 #
 # python setup.py install
@@ -153,6 +141,5 @@ def run(self):
     package_data={"graphscope.flex.rest": ["py.typed"]},
     cmdclass={
         "generate_flex_sdk": GenerateFlexSDK,
-        "bdist_wheel": CustomBDistWheel,
     },
 )
diff --git a/python/setup_gsctl.py b/python/setup_gsctl.py
index 82e2dc633eea..7a1d1e1425ca 100644
--- a/python/setup_gsctl.py
+++ b/python/setup_gsctl.py
@@ -51,7 +51,7 @@ def parse_version(root, **kwargs):
         if platform.system() == "Linux" and platform.machine() == "aarch64"
         else "click >= 8.1.6"
     ),
-    "graphscope-flex >= 0.28.0",
+    "graphscope-flex >= 0.28.1",
     "treelib",
     "packaging",
     "pyyaml",