Skip to content

Commit 1319855

Browse files
adds block sampling
Signed-off-by: Steven <skothenhill@nvidia.com>
1 parent 21b1442 commit 1319855

File tree

3 files changed

+623
-0
lines changed

3 files changed

+623
-0
lines changed

sub-packages/bionemo-core/src/bionemo/core/data/permute.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,8 +14,10 @@
1414
# limitations under the License.
1515

1616
import warnings
17+
from functools import lru_cache
1718

1819

20+
@lru_cache(maxsize=102400)
1921
def permute(index: int, length: int, seed: int) -> int:
2022
"""Index into a permuted array with constant space and time complexity.
2123

0 commit comments

Comments
 (0)