remove subsample from oc22 dataset

FAIR-Chem · Jul 29, 2024 · 65b5a6f · 65b5a6f
1 parent 235965f
commit 65b5a6f
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/src/fairchem/core/datasets/oc22_lmdb_dataset.py b/src/fairchem/core/datasets/oc22_lmdb_dataset.py
@@ -111,11 +111,8 @@ def __init__(self, config, transform=None) -> None:
         if self.train_on_oc20_total_energies:
             with open(config["oc20_ref"], "rb") as fp:
                 self.oc20_ref = pickle.load(fp)
-        self.subsample = aii(self.config.get("subsample", False), bool)
 
     def __len__(self) -> int:
-        if self.subsample:
-            return min(self.subsample, self.num_samples)
         return self.num_samples
 
     def __getitem__(self, idx):