From aab9f1fcbc677a83b698b4132a1aa04c0c20bb91 Mon Sep 17 00:00:00 2001
From: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
Date: Fri, 20 Aug 2021 15:37:48 -0400
Subject: [PATCH 1/2] fix a bug in `setup.py`

I got what's wrong here...
---
 .github/workflows/test_python.yml | 9 +++++----
 setup.py                          | 8 +++-----
 2 files changed, 8 insertions(+), 9 deletions(-)

diff --git a/.github/workflows/test_python.yml b/.github/workflows/test_python.yml
index 14cdd4cea0..8517560fc9 100644
--- a/.github/workflows/test_python.yml
+++ b/.github/workflows/test_python.yml
@@ -38,16 +38,16 @@ jobs:
             tf: 1.14
           - python: 3.7
             gcc: 5
-            tf: 2.*
+            tf:
           - python: 3.7
             gcc: 8
-            tf: 2.*
+            tf:
           - python: 3.8
             gcc: 5
-            tf: 2.*
+            tf:
           - python: 3.8
             gcc: 8
-            tf: 2.*
+            tf:
 
     steps:
     - uses: actions/checkout@master
@@ -70,4 +70,5 @@ jobs:
         CC: gcc-${{ matrix.gcc }}
         CXX: g++-${{ matrix.gcc }}
         TENSORFLOW_VERSION: ${{ matrix.tf }}
+    - run: dp --version
     - run: pytest --cov=deepmd source/tests && codecov
diff --git a/setup.py b/setup.py
index 54a9b36832..edf1377f0b 100644
--- a/setup.py
+++ b/setup.py
@@ -72,11 +72,9 @@
     # TypeError if submodule_search_locations are None
     # IndexError if submodule_search_locations is an empty list
 except (AttributeError, TypeError, IndexError):
-    setup_requires.append(f"tensorflow=={tf_version}")
-    dist = Distribution(
-        project_name="tensorflow", version=tf_version, platform=get_platform()
-    ).egg_name()
-    tf_install_dir = Path(__file__).parent.resolve().joinpath(".egg", dist, "tensorflow").resolve()
+    setup_requires.extend(extras_require['cpu'])
+    # setuptools will re-find tensorflow after installing setup_requires
+    tf_install_dir = None
 
 # add cmake as a build requirement if cmake>3.7 is not installed
 try:

From ad5e6e8c60ae2adba25ca25c9e7e47e553659877 Mon Sep 17 00:00:00 2001
From: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
Date: Fri, 20 Aug 2021 15:41:35 -0400
Subject: [PATCH 2/2] remove `Distribution`

---
 setup.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/setup.py b/setup.py
index edf1377f0b..e08f587e45 100644
--- a/setup.py
+++ b/setup.py
@@ -8,7 +8,6 @@
 from sysconfig import get_path
 
 from packaging.specifiers import SpecifierSet
-from pkg_resources import Distribution
 from skbuild import setup
 from skbuild.cmaker import get_cmake_version
 from skbuild.exceptions import SKBuildError