linting again

cellarium-ai · Jan 17, 2025 · 18bd7a5 · 18bd7a5
1 parent 481e83a
commit 18bd7a5
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/cellarium/ml/callbacks/prediction_writer.py b/cellarium/ml/callbacks/prediction_writer.py
@@ -187,6 +187,4 @@ def write_on_batch_end(
         if self.sufficient_disk_space_exists is None:
             self.sufficient_disk_space_exists = self.check_disk_space(num_files=trainer.num_predict_batches[0])
             if self.sufficient_disk_space_exists is False:
-                raise RuntimeError(
-                    f"Insufficient disk space at {self.output_dir} to write all predictions"
-                )
+                raise RuntimeError(f"Insufficient disk space at {self.output_dir} to write all predictions")