lint

Author: Jackmin801

src/lib.rs

@@ -225,7 +225,12 @@ struct Lighthouse {
 #[pymethods]
 impl Lighthouse {
     #[new]
-    fn new(py: Python<'_>, bind: String, min_replicas: u64, join_timeout_ms: Option<u64>) -> PyResult<Self> {
+    fn new(
+        py: Python<'_>,
+        bind: String,
+        min_replicas: u64,
+        join_timeout_ms: Option<u64>,
+    ) -> PyResult<Self> {
         let join_timeout_ms = join_timeout_ms.unwrap_or(100);
 
         py.allow_threads(move || {

torchft/torchft.pyi

@@ -23,6 +23,6 @@ class Manager:
     def shutdown(self) -> None: ...
 
 class Lighthouse:
-    def __init__(self, bind: str, min_replicas: int, join_timeout_ms: Optional[int]) -> None: ...
+    def __init__(self, bind: str, min_replicas: int, join_timeout_ms: Optional[int] = None) -> None: ...
     def address(self) -> str: ...
     def shutdown(self) -> None: ...