Skip to content

Commit 07a7e4b

Browse files
gitttt-1234claude
andcommitted
Fix Black formatting for lint CI
Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
1 parent cc03bf6 commit 07a7e4b

File tree

3 files changed

+29
-15
lines changed

3 files changed

+29
-15
lines changed

sleap_nn/cli.py

Lines changed: 10 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -763,13 +763,19 @@ def system():
763763
)
764764
@click.option("--backbone", type=str, help="Override backbone architecture")
765765
@click.option("--batch-size", type=int, help="Override batch size")
766-
@click.option(
767-
"-i", "--interactive", is_flag=True, help="Launch interactive TUI mode"
768-
)
766+
@click.option("-i", "--interactive", is_flag=True, help="Launch interactive TUI mode")
769767
@click.option("--analyze-only", is_flag=True, help="Only show dataset analysis")
770768
@click.option("--show-yaml", is_flag=True, help="Print YAML to stdout")
771769
def config(
772-
slp_path, output, view, pipeline, backbone, batch_size, interactive, analyze_only, show_yaml
770+
slp_path,
771+
output,
772+
view,
773+
pipeline,
774+
backbone,
775+
batch_size,
776+
interactive,
777+
analyze_only,
778+
show_yaml,
773779
):
774780
"""Generate training configuration from SLP file.
775781

sleap_nn/config_generator/recommender.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -163,9 +163,7 @@ def recommend_pipeline(stats: DatasetStats) -> PipelineRecommendation:
163163
)
164164
else:
165165
if stats.num_edges == 0:
166-
warnings.append(
167-
"No edges in skeleton - bottom-up requires edges for PAFs"
168-
)
166+
warnings.append("No edges in skeleton - bottom-up requires edges for PAFs")
169167
return PipelineRecommendation(
170168
recommended="centroid",
171169
reason="No skeleton edges available for bottom-up",
@@ -241,7 +239,9 @@ def _recommend_scale(stats: DatasetStats) -> Tuple[float, str]:
241239
return (1.0, "Image size suitable for full resolution")
242240

243241

244-
def _recommend_batch_size(stats: DatasetStats, backbone: BackboneType) -> Tuple[int, str]:
242+
def _recommend_batch_size(
243+
stats: DatasetStats, backbone: BackboneType
244+
) -> Tuple[int, str]:
245245
"""Recommend batch size based on estimated memory usage.
246246
247247
Args:

sleap_nn/config_generator/tui/app.py

Lines changed: 15 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -266,7 +266,9 @@ def _compose_data_tab(self) -> ComposeResult:
266266
with RadioSet(id="view-type"):
267267
yield RadioButton("Side View (±15° rotation)", id="view-side")
268268
yield RadioButton("Top View (±180° rotation)", id="view-top")
269-
yield RadioButton("Unknown (conservative defaults)", id="view-unknown", value=True)
269+
yield RadioButton(
270+
"Unknown (conservative defaults)", id="view-unknown", value=True
271+
)
270272

271273
yield Rule()
272274

@@ -399,9 +401,9 @@ async def on_mount(self) -> None:
399401
self.query_one("#memory-gauge", MemoryGauge).update_estimate(mem)
400402

401403
# Update recommendation panel
402-
self.query_one("#recommendation-panel", RecommendationPanel).update_recommendation(
403-
self._recommendation
404-
)
404+
self.query_one(
405+
"#recommendation-panel", RecommendationPanel
406+
).update_recommendation(self._recommendation)
405407

406408
# Initial YAML preview
407409
self._update_yaml_preview()
@@ -567,13 +569,19 @@ def _update_ui_from_generator(self) -> None:
567569
self.query_one("#sigma-input", Input).value = str(self.generator._sigma)
568570

569571
# Output stride
570-
self.query_one("#output-stride-select", Select).value = self.generator._output_stride
572+
self.query_one("#output-stride-select", Select).value = (
573+
self.generator._output_stride
574+
)
571575

572576
# Batch size
573-
self.query_one("#batch-size-input", Input).value = str(self.generator._batch_size)
577+
self.query_one("#batch-size-input", Input).value = str(
578+
self.generator._batch_size
579+
)
574580

575581
# Max epochs
576-
self.query_one("#max-epochs-input", Input).value = str(self.generator._max_epochs)
582+
self.query_one("#max-epochs-input", Input).value = str(
583+
self.generator._max_epochs
584+
)
577585

578586
# Learning rate
579587
self.query_one("#learning-rate-input", Input).value = str(

0 commit comments

Comments
 (0)