Merge pull request #87 from xadrianzetx/support-py-312

Explicitly support Python 3.12 + reformat codebase with `black 24.x`

.github/workflows/test.yaml

@@ -24,6 +24,7 @@ jobs:
           - '3.9'
           - '3.10'
           - '3.11'
+          - '3.12'
 
     steps:
     - uses: actions/checkout@v3

optuna_distributed/managers/distributed.py

@@ -38,8 +38,7 @@
 DistributableWithContext = Callable[["_TaskContext"], None]
 
 
-class WorkerInterrupted(Exception):
-    ...
+class WorkerInterrupted(Exception): ...
 
 
 class _TaskState(IntEnum):
@@ -166,8 +165,7 @@ def get_message(self) -> Generator[Message, None, None]:
                 # e.g. in `after_message`.
                 yield HeartbeatMessage()
 
-    def after_message(self, event_loop: "EventLoop") -> None:
-        ...
+    def after_message(self, event_loop: "EventLoop") -> None: ...
 
     def get_connection(self, trial_id: int) -> IPCPrimitive:
         return Queue(self._private_channels[trial_id])

optuna_distributed/messages/heartbeat.py

@@ -19,5 +19,4 @@ class HeartbeatMessage(Message):
 
     closing = False
 
-    def process(self, study: Study, manager: "OptimizationManager") -> None:
-        ...
+    def process(self, study: Study, manager: "OptimizationManager") -> None: ...

optuna_distributed/messages/response.py

@@ -36,5 +36,4 @@ def __init__(self, trial_id: int, data: T) -> None:
         self.trial_id = trial_id
         self.data = data
 
-    def process(self, study: Study, manager: "OptimizationManager") -> None:
-        ...
+    def process(self, study: Study, manager: "OptimizationManager") -> None: ...

pyproject.toml

@@ -20,6 +20,7 @@ classifiers = [
   "Programming Language :: Python :: 3.9",
   "Programming Language :: Python :: 3.10",
   "Programming Language :: Python :: 3.11",
+  "Programming Language :: Python :: 3.12",
   "Programming Language :: Python :: 3 :: Only",
   "Topic :: Scientific/Engineering",
   "Topic :: Scientific/Engineering :: Mathematics",

setup.cfg

@@ -1,5 +1,5 @@
 [flake8]
-ignore = D203
+ignore = D203, E701, E704
 exclude =
     .git,
     __pycache__,

tests/test_messages.py

@@ -43,32 +43,27 @@ def put(self, message: Message) -> None:
         assert isinstance(message, ResponseMessage)
         self._manager.message_response = message.data
 
-    def close(self) -> None:
-        ...
+    def close(self) -> None: ...
 
 
 class MockOptimizationManager(OptimizationManager):
     def __init__(self) -> None:
         self.trial_exit_called = False
         self.message_response = None
 
-    def create_futures(self, study: "Study", objective: ObjectiveFuncType) -> None:
-        ...
+    def create_futures(self, study: "Study", objective: ObjectiveFuncType) -> None: ...
 
-    def before_message(self, event_loop: "EventLoop") -> None:
-        ...
+    def before_message(self, event_loop: "EventLoop") -> None: ...
 
     def get_message(self) -> Generator["Message", None, None]:
         yield HeartbeatMessage()
 
-    def after_message(self, event_loop: "EventLoop") -> None:
-        ...
+    def after_message(self, event_loop: "EventLoop") -> None: ...
 
     def get_connection(self, trial_id: int) -> "IPCPrimitive":
         return MockConnection(self)
 
-    def stop_optimization(self, patience: float) -> None:
-        ...
+    def stop_optimization(self, patience: float) -> None: ...
 
     def should_end_optimization(self) -> bool:
         return True

tests/test_trial.py

@@ -29,8 +29,7 @@ def get(self) -> "Message":
     def put(self, message: "Message") -> None:
         self.captured.append(message)
 
-    def close(self) -> None:
-        ...
+    def close(self) -> None: ...
 
     def enqueue_response(self, response: ResponseMessage) -> None:
         self.responses.append(response)