[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

jupyter_scheduler/executors.py

@@ -29,9 +29,7 @@ class ExecutionManager(ABC):
     _model = None
     _db_session = None
 
-    def __init__(
-        self, job_id: str, root_dir: str, db_url: str, staging_paths: Dict[str, str]
-    ):
+    def __init__(self, job_id: str, root_dir: str, db_url: str, staging_paths: Dict[str, str]):
         self.job_id = job_id
         self.staging_paths = staging_paths
         self.root_dir = root_dir
@@ -175,14 +173,10 @@ def add_side_effects_files(self, staging_dir: str):
         if new_files_set:
             with self.db_session() as session:
                 current_packaged_files_set = set(
-                    session.query(Job.packaged_files)
-                    .filter(Job.job_id == self.job_id)
-                    .scalar()
+                    session.query(Job.packaged_files).filter(Job.job_id == self.job_id).scalar()
                     or []
                 )
-                updated_packaged_files = list(
-                    current_packaged_files_set.union(new_files_set)
-                )
+                updated_packaged_files = list(current_packaged_files_set.union(new_files_set))
                 session.query(Job).filter(Job.job_id == self.job_id).update(
                     {"packaged_files": updated_packaged_files}
                 )
@@ -192,9 +186,7 @@ def create_output_files(self, job: DescribeJob, notebook_node):
         for output_format in job.output_formats:
             cls = nbconvert.get_exporter(output_format)
             output, _ = cls().from_notebook_node(notebook_node)
-            with fsspec.open(
-                self.staging_paths[output_format], "w", encoding="utf-8"
-            ) as f:
+            with fsspec.open(self.staging_paths[output_format], "w", encoding="utf-8") as f:
                 f.write(output)
 
     def supported_features(self) -> Dict[JobFeature, bool]:

jupyter_scheduler/tests/test_execution_manager.py

@@ -157,8 +157,8 @@ def test_update_completed_cells_hook_database_error():
     # Mock db_session with error
     mock_db_session = MagicMock()
     mock_session_context = MagicMock()
-    mock_session_context.query.return_value.filter.return_value.update.side_effect = (
-        Exception("DB Error")
+    mock_session_context.query.return_value.filter.return_value.update.side_effect = Exception(
+        "DB Error"
     )
     mock_db_session.return_value.__enter__.return_value = mock_session_context
     manager._db_session = mock_db_session
@@ -260,12 +260,8 @@ def supported_features(self):
         with patch.object(DisabledTrackingExecutionManager, "model") as mock_model:
             with patch("jupyter_scheduler.executors.open", mock=MagicMock()):
                 with patch("jupyter_scheduler.executors.nbformat.read") as mock_nb_read:
-                    with patch.object(
-                        DisabledTrackingExecutionManager, "add_side_effects_files"
-                    ):
-                        with patch.object(
-                            DisabledTrackingExecutionManager, "create_output_files"
-                        ):
+                    with patch.object(DisabledTrackingExecutionManager, "add_side_effects_files"):
+                        with patch.object(DisabledTrackingExecutionManager, "create_output_files"):
                             with patch(
                                 "jupyter_scheduler.executors.ExecutePreprocessor"
                             ) as mock_ep_class: