[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

@@ -30,24 +30,30 @@ 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], database_manager_class):
+    def __init__(
+        self,
+        job_id: str,
+        root_dir: str,
+        db_url: str,
+        staging_paths: Dict[str, str],
+        database_manager_class,
+    ):
         self.job_id = job_id
         self.staging_paths = staging_paths
         self.root_dir = root_dir
         self.db_url = db_url
-        
+
         self.database_manager = self._create_database_manager(database_manager_class)
 
-    
     def _create_database_manager(self, database_manager_class):
         try:
-            module_name, class_name = database_manager_class.rsplit('.', 1)
+            module_name, class_name = database_manager_class.rsplit(".", 1)
             module = importlib.import_module(module_name)
             DatabaseManagerClass = getattr(module, class_name)
             return DatabaseManagerClass()
         except (ValueError, ImportError, AttributeError) as e:
             raise ValueError(f"Invalid database_manager_class '{database_manager_class}': {e}")
-    
+
     @property
     def model(self):
         if self._model is None:

jupyter_scheduler/managers.py

@@ -1,35 +1,37 @@
 from abc import ABC, abstractmethod
 from sqlite3 import OperationalError
+
 from sqlalchemy import create_engine
 from sqlalchemy.orm import sessionmaker
 
-from jupyter_scheduler.orm import Base as DefaultBase, update_db_schema
+from jupyter_scheduler.orm import Base as DefaultBase
+from jupyter_scheduler.orm import update_db_schema
 
 
 class DatabaseManager(ABC):
     """Base class for database managers.
-    
+
     Database managers handle database operations for jupyter-scheduler.
     Subclasses can implement custom storage backends (K8s, Redis, etc.)
     while maintaining compatibility with the scheduler's session interface.
     """
-    
+
     @abstractmethod
     def create_session(self, db_url: str):
         """Create a database session.
-        
+
         Args:
             db_url: Database URL (e.g., "k8s://namespace", "redis://localhost")
-            
+
         Returns:
             Session object compatible with SQLAlchemy session interface
         """
         pass
-    
-    @abstractmethod  
+
+    @abstractmethod
     def create_tables(self, db_url: str, drop_tables: bool = False, Base=None):
         """Create database tables/schema.
-        
+
         Args:
             db_url: Database URL
             drop_tables: Whether to drop existing tables first
@@ -40,25 +42,25 @@ def create_tables(self, db_url: str, drop_tables: bool = False, Base=None):
 
 class SQLAlchemyDatabaseManager(DatabaseManager):
     """Default database manager using SQLAlchemy."""
-    
+
     def create_session(self, db_url: str):
         """Create SQLAlchemy session factory."""
         engine = create_engine(db_url, echo=False)
         Session = sessionmaker(bind=engine)
         return Session
-    
+
     def create_tables(self, db_url: str, drop_tables: bool = False, Base=None):
         """Create database tables using SQLAlchemy."""
         if Base is None:
             Base = DefaultBase
-        
+
         engine = create_engine(db_url)
         update_db_schema(engine, Base)
-        
+
         try:
             if drop_tables:
                 Base.metadata.drop_all(engine)
         except OperationalError:
             pass
         finally:
-            Base.metadata.create_all(engine)
+            Base.metadata.create_all(engine)

jupyter_scheduler/tests/test_orm.py

@@ -15,11 +15,14 @@
 @pytest.fixture
 def database_manager():
     from jupyter_scheduler.managers import SQLAlchemyDatabaseManager
+
     return SQLAlchemyDatabaseManager()
 
 
 @pytest.fixture
-def initial_db(jp_scheduler_db_url, database_manager) -> tuple[Type[DeclarativeMeta], sessionmaker, str]:
+def initial_db(
+    jp_scheduler_db_url, database_manager
+) -> tuple[Type[DeclarativeMeta], sessionmaker, str]:
     TestBase = declarative_base()
 
     class MockInitialJob(TestBase):
@@ -58,7 +61,9 @@ class MockUpdatedJob(TestBase):
     return MockUpdatedJob
 
 
-def test_create_tables_with_new_column(jp_scheduler_db_url, initial_db, updated_job_model, database_manager):
+def test_create_tables_with_new_column(
+    jp_scheduler_db_url, initial_db, updated_job_model, database_manager
+):
     TestBase, Session, initial_job_id = initial_db
 
     session = Session()