From 7868df27dc168a2e1de11d86983184ce8eb88a97 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Wed, 3 Sep 2025 14:18:02 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- src/lerobot/datasets/lerobot_dataset.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/src/lerobot/datasets/lerobot_dataset.py b/src/lerobot/datasets/lerobot_dataset.py index 9036827fa..2bd277b3b 100644 --- a/src/lerobot/datasets/lerobot_dataset.py +++ b/src/lerobot/datasets/lerobot_dataset.py @@ -315,13 +315,13 @@ class LeRobotDatasetMetadata: # Update the Hugging Face dataset by reloading it. # This process should be fast because only the latest Parquet file has been modified. # Therefore, only this file needs to be converted to PyArrow; the rest is loaded from the PyArrow memory-mapped cache. - + # Explicitly delete old dataset to free memory before reloading - if hasattr(self, 'episodes') and self.episodes is not None: + if hasattr(self, "episodes") and self.episodes is not None: del self.episodes self.episodes = None - gc.collect() - + gc.collect() + self.episodes = load_episodes(self.root) def save_episode( @@ -1059,13 +1059,13 @@ class LeRobotDataset(torch.utils.data.Dataset): # Update the Hugging Face dataset by reloading it. # This process should be fast because only the latest Parquet file has been modified. # Therefore, only this file needs to be converted to PyArrow; the rest is loaded from the PyArrow memory-mapped cache. - + # Explicitly delete old dataset to free memory before reloading - if hasattr(self, 'hf_dataset') and self.hf_dataset is not None: + if hasattr(self, "hf_dataset") and self.hf_dataset is not None: del self.hf_dataset self.hf_dataset = None gc.collect() - + self.hf_dataset = self.load_hf_dataset() metadata = {