diff --git a/.github/workflows/cpp_testing.yml b/.github/workflows/cpp_testing.yml index 5e5c28f7b..e6ad6b892 100644 --- a/.github/workflows/cpp_testing.yml +++ b/.github/workflows/cpp_testing.yml @@ -23,7 +23,7 @@ jobs: - name: Configure run: | export MKL_VERSION=`python -c 'from tools.mkl_ver import compatible_mkl_ver;print(compatible_mkl_ver())'` - pip install mkl-include==${MKL_VERSION} mkl-static==${MKL_VERSION} + pip install --no-cache-dir mkl-include==${MKL_VERSION} mkl-static==${MKL_VERSION} export _BLAS_INCLUDE_DIR=`python -c 'import os;import sysconfig;data=sysconfig.get_path("data");print(f"{data}{os.sep}include")'` export MKL_DIR=`python -c 'import os;import sysconfig;data=sysconfig.get_path("data");print(f"{data}{os.sep}lib{os.sep}cmake{os.sep}mkl")'` diff --git a/benchmark/sampler/subgraph.py b/benchmark/sampler/subgraph.py index a5734baeb..3ace1a779 100644 --- a/benchmark/sampler/subgraph.py +++ b/benchmark/sampler/subgraph.py @@ -16,7 +16,7 @@ def test_subgraph(dataset, **kwargs): t = time.perf_counter() for _ in range(10): pyg_lib.sampler.subgraph(rowptr, col, nodes) - print(f'time={time.perf_counter()-t:.6f} seconds') + print(f'time={time.perf_counter() - t:.6f} seconds') edge_index = to_edge_index(rowptr, col) from torch_geometric.utils import subgraph @@ -24,7 +24,7 @@ def test_subgraph(dataset, **kwargs): t = time.perf_counter() for _ in range(10): subgraph(nodes, edge_index, num_nodes=num_nodes, relabel_nodes=True) - print(f'time={time.perf_counter()-t:.6f} seconds') + print(f'time={time.perf_counter() - t:.6f} seconds') if __name__ == '__main__':