Skip to content

Commit

Permalink
linter is still complaining......
Browse files Browse the repository at this point in the history
  • Loading branch information
akrzgc committed Oct 27, 2023
1 parent e896274 commit 8899158
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions test/test_benchmark_performance.py
Original file line number Diff line number Diff line change
@@ -1,9 +1,9 @@
# Copyright (c) 2023 Graphcore Ltd. All rights reserved.
from subprocess import Popen, call

import jax
import numpy as np
import pytest

from tessellate_ipu import (
ipu_cycle_count,
tile_map,
Expand Down Expand Up @@ -41,11 +41,11 @@ def jitted_inner_test(dummy, dummier):

assert True


@pytest.mark.skipif(not has_ipu(), reason="Skipping ipu test!")
@pytest.mark.ipu
@pytest.mark.parametrize("molecule", ["methane", "benzene"])
def test_dense_eri(molecule):

opts, mol_str = nanoDFT_options(float32=True, mol_str=molecule, backend="ipu")
mol = build_mol(mol_str, opts.basis)

Expand All @@ -72,7 +72,6 @@ def test_dense_eri(molecule):
@pytest.mark.ipu
@pytest.mark.parametrize("molecule", ["methane", "benzene", "c20"])
def test_sparse_eri(molecule):

opts, mol_str = nanoDFT_options(
float32=True,
mol_str=molecule,
Expand Down

0 comments on commit 8899158

Please sign in to comment.