fixing setup.cfg

    - name: Build DeepChem

      run: |

        python -m pip install --upgrade pip

        pip install tensorflow'>=2.3,<2.4'

        pip install tensorflow==2.4

        pip install -e '.[jax]'

    - name: Import checking

      run: python -c "import deepchem; import jax;"

    - name: PyTest

      if: ${{ (success() || failure()) && (steps.install.outcome == 'failure' || steps.install.outcome == 'success') }}

      shell: bash -l {0}

      run: pytest -v -m jax

      run: pytest -v -m jax deepchem

"""

checking jax imports for new CI build

"""

import deepchem as dc

import pytest

try:

  import jax.numpy as jnp

  from jax import random

  import numpy as np

import deepchem as dc

import pytest

  has_jax = True

except:

  has_jax = False

@pytest.mark.jax

[tool:pytest]

markers =

    jax: marks tests for deepchem-jax only (deselect with '-m "not jax"')

    slow: marks tests as slow (deselect with '-m "not slow"')

    serial