google / flax
Showing 1 of 1 files from the diff.

@@ -18,6 +18,8 @@
Loading
18 18
import functools
19 19
import hashlib
20 20
import dataclasses
21 +
22 +
import typing
21 23
from typing import Any, Callable, Container, Dict, Generic, Iterable, Mapping, Optional, Sequence, Set, Tuple, TypeVar, Union
22 24
23 25
from . import tracers
@@ -113,7 +115,7 @@
Loading
113 115
  return rng
114 116
115 117
116 -
def _fold_in_static(rng: PRNGKey, data: Iterable[PRNGFoldable]) -> PRNGKey:
118 +
def _fold_in_static(rng: PRNGKey, data: typing.Collection[PRNGFoldable]) -> PRNGKey:
117 119
  """Folds static data (strings & ints) into a jax.random.PRNGKey using its SHA-1 hash.
118 120
119 121
  This is faster than splitting an PRNGKey because it allows generating new PRNG
Files Coverage
flax 83.18%
Project Totals (69 files) 83.18%
Notifications are pending CI completion. Periodically Codecov will check the CI state, when complete notifications will be submitted. Push notifications now.

No yaml found.

Create your codecov.yml to customize your Codecov experience

Sunburst
The inner-most circle is the entire project, moving away from the center are folders then, finally, a single file. The size and color of each slice is representing the number of statements and the coverage, respectively.
Icicle
The top section represents the entire project. Proceeding with folders and finally individual files. The size and color of each slice is representing the number of statements and the coverage, respectively.
Grid
Each block represents a single file in the project. The size and color of each block is represented by the number of statements and the coverage, respectively.
Loading