Fixed pdb dataset init

a-r-j · Mar 13, 2024 · 1115a57 · 1115a57
1 parent 2df0aad
commit 1115a57
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/proteinworkshop/datasets/pdb_dataset.py b/proteinworkshop/datasets/pdb_dataset.py
@@ -14,7 +14,7 @@
 from proteinworkshop.datasets.utils import download_pdb_mmtf
 
 class PDBData:
-    def _init_(
+    def __init__(
         self,
         fraction: float,
         min_length: int,
@@ -128,7 +128,7 @@ def create_dataset(self):
 
 
 class PDBDataModule(ProteinDataModule):
-    def _init_(
+    def __init__(
         self,
         path: Optional[str] = None,
         structure_dir: Optional[str] = None,
@@ -141,7 +141,7 @@ def _init_(
         structure_format: str = "mmtf.gz",
         overwrite: bool = False,
     ):
-        super()._init_()
+        super().__init__()
         self.root = path
         self.dataset = pdb_dataset
         self.dataset.path = path