Mark operation evaluated after the op is run (#1528)

alibaba · May 7, 2022 · c03b539 · c03b539
1 parent 4c16d53
commit c03b539
Show file tree

Hide file tree

Showing 4 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -484,6 +484,7 @@ jobs:
             - 'analytical_engine/apps/**'
             - 'analytical_engine/frame/**'
             - 'python/graphscope/nx/algorithms/builtin.py'
+            - 'python/graphscope/nx/algorithms/tests/builtin/**'
           generator:
             - 'python/graphscope/nx/classes/**'
             - 'python/graphscope/nx/utils/**'

diff --git a/python/graphscope/framework/operation.py b/python/graphscope/framework/operation.py
@@ -150,6 +150,7 @@ def eval(self, leaf=True):
         if not self._leaf:
             sess.dag.add_op(self)
         res = sess.run(self)
+        self._evaluated = True
         return res
 
     def add_parent(self, op):

diff --git a/python/graphscope/nx/algorithms/tests/builtin/test_eigenvector_centrality.py b/python/graphscope/nx/algorithms/tests/builtin/test_eigenvector_centrality.py
@@ -67,7 +67,6 @@ def test_maxiter(self):
             b = nx.builtin.eigenvector_centrality(G, max_iter=0)
 
 
-@pytest.mark.skip(reason="GetObject -> Object does not exist")
 @pytest.mark.usefixtures("graphscope_session")
 class TestEigenvectorCentralityDirected(object):
     @classmethod

diff --git a/python/graphscope/nx/algorithms/tests/builtin/test_katz_centrality.py b/python/graphscope/nx/algorithms/tests/builtin/test_katz_centrality.py
@@ -269,7 +269,6 @@ def test_P3_unweighted(self):
             assert almost_equal(b[n], b_answer[n], places=4)
 
 
-@pytest.mark.skip(reason="GetObject -> Object does not exist")
 @pytest.mark.usefixtures("graphscope_session")
 class TestKatzCentralityDirected(object):
     @classmethod