amend

pytorch · Jan 16, 2024 · 600d9f4 · 600d9f4
1 parent b3cd4f3
commit 600d9f4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/torchrl/data/datasets/gen_dgrl.py b/torchrl/data/datasets/gen_dgrl.py
@@ -238,7 +238,9 @@ def _unpack_category_file(
             batch = self._PROCESS_NPY_BATCH
         _, file_name, _ = link
         file_path = os.path.join(download_folder, file_name)
-        logging.info(f"Unpacking dataset file {file_path} ({file_name}) to {download_folder}.")
+        logging.info(
+            f"Unpacking dataset file {file_path} ({file_name}) to {download_folder}."
+        )
         idx = 0
         td_memmap = None
         dataset_len = self._get_category_len(category_name)