<commit_before>language: python
python:
  - "3.6"
  - "3.7"
  - "3.8"
install:
  - pip install --upgrade pip
  - pip install numpy jaxlib jax tensorflow tensorflow-datasets --upgrade
  - pip install -e .
script:
  - set -e
  - for f in tests/*.py; do python $f; done<commit_msg>Use JAX at github head instead of pypi, to fix failing tests<commit_after>language: python
python:
  - "3.6"
  - "3.7"
  - "3.8"
install:
  - pip install --upgrade pip
  - pip install numpy jaxlib tensorflow tensorflow-datasets --upgrade
  - pip install git+https://github.com/google/jax.git
  - pip install -e .
script:
  - set -e
  - for f in tests/*.py; do python $f; done

{'fraction_non_alphanumeric': 0.13149847094801223, 'fraction_numerical': 0.01834862385321101, 'mean_word_length': 2.6187845303867405, 'pattern_counts': {'":': 0, '<': 3, '<?xml version=': 0, '>': 3, 'https://': 1, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '3038347', 'n_tokens_mistral': 244, 'n_tokens_neox': 237, 'n_words': 78}