Merge pull request #7124 from jakevdp:fix-flake

PiperOrigin-RevId: 381913589
This commit is contained in:
jax authors 2021-06-28 11:51:45 -07:00
commit 0d68dbd619

View File

@ -48,7 +48,7 @@ DeviceArray([-1.2655463 , -0.52060574, -0.14522289, -0.10817424,
import functools
from typing import (
Any, Callable, Dict, NamedTuple, NewType, List, Optional, Sequence, Tuple, Union)
Any, Callable, Dict, NamedTuple, List, Optional, Sequence, Tuple, Union)
import numpy as np