Skip to content

Commit ed1708d

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent c34c428 commit ed1708d

File tree

4 files changed

+16
-20
lines changed

4 files changed

+16
-20
lines changed

jupyter_scheduler/backend_registry.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,6 @@
1616
from jupyter_scheduler.orm import create_tables
1717
from jupyter_scheduler.scheduler import BaseScheduler
1818

19-
2019
logger = logging.getLogger(__name__)
2120

2221

@@ -98,9 +97,7 @@ def initialize(
9897
"""
9998
for cfg in self._configs:
10099
try:
101-
instance = self._create_backend(
102-
cfg, root_dir, environments_manager, db_url, config
103-
)
100+
instance = self._create_backend(cfg, root_dir, environments_manager, db_url, config)
104101
self._backends[cfg.id] = instance
105102

106103
# Build extension map for auto-selection

jupyter_scheduler/backends.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
"""
1010

1111
from dataclasses import dataclass, field
12-
from typing import Dict, List, Optional, Any
12+
from typing import Any, Dict, List, Optional
1313

1414
from jupyter_scheduler.pydantic_v1 import BaseModel
1515

jupyter_scheduler/extension.py

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,9 @@
44
from jupyter_core.paths import jupyter_data_dir
55
from jupyter_server.extension.application import ExtensionApp
66
from jupyter_server.transutils import _i18n
7-
from traitlets import Bool, List as TList, Type, Unicode, default
7+
from traitlets import Bool
8+
from traitlets import List as TList
9+
from traitlets import Type, Unicode, default
810

911
from jupyter_scheduler.backend_registry import BackendRegistry
1012
from jupyter_scheduler.backends import BackendConfig
@@ -74,9 +76,7 @@ def _default_backends(self):
7476
default_value="jupyter_scheduler.scheduler.Scheduler",
7577
klass="jupyter_scheduler.scheduler.BaseScheduler",
7678
config=True,
77-
help=_i18n(
78-
"The scheduler class to use. Deprecated: use 'backends' configuration instead."
79-
),
79+
help=_i18n("The scheduler class to use. Deprecated: use 'backends' configuration instead."),
8080
)
8181

8282
job_files_manager_class = Type(
@@ -134,9 +134,7 @@ def initialize_settings(self):
134134
backend_configs = self._build_backend_configs()
135135

136136
# Determine default backend
137-
default_id = next(
138-
(c.id for c in backend_configs if c.is_default), backend_configs[0].id
139-
)
137+
default_id = next((c.id for c in backend_configs if c.is_default), backend_configs[0].id)
140138

141139
# Create and initialize the backend registry
142140
registry = BackendRegistry(backend_configs, default_id)

jupyter_scheduler/tests/test_backend_registry.py

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,15 @@
11
"""Unit tests for BackendRegistry."""
22

3-
import pytest
43
from unittest.mock import MagicMock, patch
54

5+
import pytest
6+
7+
from jupyter_scheduler.backend_registry import (
8+
BackendInstance,
9+
BackendRegistry,
10+
import_class,
11+
)
612
from jupyter_scheduler.backends import BackendConfig, DescribeBackend
7-
from jupyter_scheduler.backend_registry import BackendRegistry, BackendInstance, import_class
813

914

1015
class TestImportClass:
@@ -298,9 +303,7 @@ def test_list_backends_returns_describe_backend_model(
298303

299304
@patch("jupyter_scheduler.backend_registry.create_tables")
300305
@patch("jupyter_scheduler.backend_registry.import_class")
301-
def test_contains_operator(
302-
self, mock_import, mock_create_tables, local_backend_config
303-
):
306+
def test_contains_operator(self, mock_import, mock_create_tables, local_backend_config):
304307
"""Contains operator works correctly."""
305308
mock_scheduler_class = MagicMock()
306309
mock_scheduler_class.return_value = MagicMock()
@@ -393,9 +396,7 @@ def test_uses_global_db_url_when_backend_has_none(self, mock_import, mock_create
393396

394397
@patch("jupyter_scheduler.backend_registry.create_tables")
395398
@patch("jupyter_scheduler.backend_registry.import_class")
396-
def test_extension_map_normalizes_extensions(
397-
self, mock_import, mock_create_tables
398-
):
399+
def test_extension_map_normalizes_extensions(self, mock_import, mock_create_tables):
399400
"""Extension map normalizes extensions (lowercase, no leading dot)."""
400401
config = BackendConfig(
401402
id="test",

0 commit comments

Comments
 (0)