Skip to content

Commit

Permalink
tests/helpers.py: fix py38 vers incompatibility from other PR
Browse files Browse the repository at this point in the history
  • Loading branch information
Titus-von-Koeller authored Feb 21, 2024
1 parent a84b660 commit 0bf7198
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/helpers.py
Original file line number Diff line number Diff line change
@@ -1,13 +1,13 @@
from itertools import product
import random
from typing import Any
from typing import Any, List

import torch

test_dims_rng = random.Random(42)


def get_test_dims(min: int, max: int, *, n: int) -> list[int]:
def get_test_dims(min: int, max: int, *, n: int) -> List[int]:
return [test_dims_rng.randint(min, max) for _ in range(n)]


Expand Down

0 comments on commit 0bf7198

Please sign in to comment.