Add networkx dep

pydata · Jun 12, 2024 · 1a22a38 · 1a22a38
1 parent 64fc60b
commit 1a22a38
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -100,7 +100,7 @@ jobs:
     - name: Build and install Sparse
       run: |
         pip install -U setuptools wheel
-        python -m pip install '.[finch]' scipy
+        python -m pip install '.[finch]' scipy networkx
     - name: Run examples
       run: |
         source ci/test_examples.sh

diff --git a/examples/pagerank_example.py b/examples/pagerank_example.py
@@ -1,3 +1,5 @@
+import importlib
+import os
 import time
 
 import sparse
@@ -54,6 +56,9 @@ def pagerank(G, alpha=0.85, max_iter=100, tol=1e-6) -> dict:
     G = nx.DiGraph(nx.path_graph(4))
     ITERS = 3
 
+    os.environ[sparse._ENV_VAR_NAME] = "Finch"
+    importlib.reload(sparse)
+
     # compile
     pagerank(G)
     print("compiled")