Running flake8

import os

import logging

import tempfile

from typing import Dict, List, Optional, Tuple, Union

from typing import Dict, List, Optional, Tuple, Union, Any

from deepchem.data import Dataset

from deepchem.trans import Transformer

    param_keys = list(param_range.keys())

    # Stores all results

    all_results = {}

    all_results: Dict[Any, Any] = {}

    # Store all model references so we don't have to reload

    all_models = {}

    # Stores all model locations

import math

import numpy as np

import logging

try:

  import jax.numpy as jnp

  import jax

  import haiku as hk

except:

  raise ImportError('These classes require Jax and Haiku to be installed.')