Skip to content

Commit e07b5ae

Browse files
committed
Fix code wrapping in _export, Slicer.__init__, and format_model
Close paren on its own line for multi-line function signatures.
1 parent a8f1b2a commit e07b5ae

3 files changed

Lines changed: 17 additions & 11 deletions

File tree

deployment/exporter.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -108,8 +108,10 @@ def __init__(
108108
self.dur2bd_path = self.save_dir / "dur2bd.onnx"
109109
self.bd2dur_path = self.save_dir / "bd2dur.onnx"
110110

111-
def _export(self, model, args, save_path, input_names, output_names,
112-
dynamic_axes=None, dynamic_shapes=None, kwargs=None):
111+
def _export(
112+
self, model, args, save_path, input_names, output_names,
113+
dynamic_axes=None, dynamic_shapes=None, kwargs=None,
114+
):
113115
if self.dynamo:
114116
program = torch.onnx.export(
115117
model, args, None,

inference/slicer2.py

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -39,13 +39,15 @@ def get_rms(
3939

4040

4141
class Slicer:
42-
def __init__(self,
43-
sr: int,
44-
threshold: float = -40.,
45-
min_length: int = 5000,
46-
min_interval: int = 300,
47-
hop_size: int = 20,
48-
max_sil_kept: int = 5000):
42+
def __init__(
43+
self,
44+
sr: int,
45+
threshold: float = -40.,
46+
min_length: int = 5000,
47+
min_interval: int = 300,
48+
hop_size: int = 20,
49+
max_sil_kept: int = 5000,
50+
):
4951
if not min_length >= min_interval >= hop_size:
5052
raise ValueError('The following condition must be satisfied: min_length >= min_interval >= hop_size')
5153
if not max_sil_kept >= hop_size:

lib/config/formatter.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -133,8 +133,10 @@ def _add_entry(fmt: _Fmt, key: Optional[str], value: Any):
133133
fmt.cur_width += width
134134

135135

136-
def format_model(model: BaseModel, line_width: int = 80, indent: int = 4,
137-
connector: str = ": ", separator: str = ", ") -> str:
136+
def format_model(
137+
model: BaseModel, line_width: int = 80, indent: int = 4,
138+
connector: str = ": ", separator: str = ", ",
139+
) -> str:
138140
"""Format a Pydantic model as a readable string."""
139141
fmt = _Fmt(line_width=line_width, indent=indent, connector=connector, separator=separator)
140142
_add_entry(fmt, None, model)

0 commit comments

Comments
 (0)