Skip to content

GitLab

Unverified Commit babce128 authored by atgctg's avatar atgctg Committed by GitHub
Browse files

Fix probabilities type (`int` -> `float`)

parent 55c5623d
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
openfold/data/data_modules.py
View file @ babce128
...@@ -334,7 +334,7 @@ class OpenFoldDataset(torch.utils.data.Dataset): ...@@ -334,7 +334,7 @@ class OpenFoldDataset(torch.utils.data.Dataset):
""" """
def __init__(self, def __init__(self,
datasets: Sequence[OpenFoldSingleDataset], datasets: Sequence[OpenFoldSingleDataset],
probabilities: Sequence[int], probabilities: Sequence[float],
epoch_len: int, epoch_len: int,
generator: torch.Generator = None, generator: torch.Generator = None,
_roll_at_init: bool = True, _roll_at_init: bool = True,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment