break line in 2 to reduce line length

dilyabareeva · Apr 18, 2024 · 424a1ca · 424a1ca
1 parent 9019ade
commit 424a1ca
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/utils/datasets/corrupt_label_dataset.py b/src/utils/datasets/corrupt_label_dataset.py
@@ -26,7 +26,8 @@ def __init__(
         assert hasattr(dataset, "device")
         self.device = dataset.device
 
-        if IC.exists(path=cache_path, file_id=f"{dataset_id}_corrupt_ids") and IC.exists(path=cache_path, file_id=f"{dataset_id}_corrupt_labels"):
+        if IC.exists(path=cache_path, file_id=f"{dataset_id}_corrupt_ids") and \
+            IC.exists(path=cache_path, file_id=f"{dataset_id}_corrupt_labels"):
             self.corrupt_indices = IC.load(path=cache_path, file_id=f"{dataset_id}_corrupt_ids")
             self.corrupt_labels = IC.load(path=cache_path, file_id=f"{dataset_id}_corrupt_labels", device=self.device)
         else: