Fix typo

torchmd · Oct 27, 2023 · d359155 · d359155
1 parent f480e43
commit d359155
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchmdnet/datasets/custom.py b/torchmdnet/datasets/custom.py
@@ -76,7 +76,7 @@ def __init__(
         print("Number of files: ", len(self.files["pos"]))
         self.cached = False
         total_data_size = self._initialize_index()
-        print(f"Combined dataset size {len(self.index))}")
+        print(f"Combined dataset size {len(self.index)}")
         # If the dataset is small enough, load it whole into CPU memory
         data_size_limit = preload_memory_limit * 1024 * 1024
         if total_data_size < data_size_limit: