Skip to content

GitLab

Unverified Commit 46e6a876 authored by Gustaf Ahdritz's avatar Gustaf Ahdritz Committed by GitHub
Browse files

Merge pull request #225 from atgctg/patch-1 

Fix probabilities type (`int` -> `float`)
parents 55c5623d babce128
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
openfold/data/data_modules.py
View file @ 46e6a876
...@@ -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