Update revision-scripts/mp_classification.py

Co-authored-by: sourcery-ai[bot] <58596630+sourcery-ai[bot]@users.noreply.github.com>
lamalab-org · Aug 14, 2024 · 0de6dda · 0de6dda
1 parent 0dfe9ee
commit 0de6dda
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/revision-scripts/mp_classification.py b/revision-scripts/mp_classification.py
@@ -21,8 +21,7 @@ def __len__(self):
 
     def get(self, index):
         id = f"{index}".encode("ascii")
-        datapoint = pickle.loads(self.txn.get(id))
-        return datapoint
+        return pickle.loads(self.txn.get(id))
 
 def create_json_from_lmdb(lmdb_path, output_dir):
     dataset = Dataset(lmdb_path)