Skip to content

Commit

Permalink
[pre-commit.ci] pre-commit autoupdate (#336)
Browse files Browse the repository at this point in the history
* [pre-commit.ci] pre-commit autoupdate

updates:
- [github.com/astral-sh/ruff-pre-commit: v0.7.4 → v0.8.0](astral-sh/ruff-pre-commit@v0.7.4...v0.8.0)

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
  • Loading branch information
pre-commit-ci[bot] authored Nov 26, 2024
1 parent 3f8e998 commit d937bbd
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 6 deletions.
2 changes: 1 addition & 1 deletion .pre-commit-config.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@ ci:
skip: [mypy, pytest]
repos:
- repo: https://github.com/astral-sh/ruff-pre-commit
rev: v0.7.4
rev: v0.8.0
hooks:
- id: ruff
args: [--fix]
Expand Down
2 changes: 1 addition & 1 deletion torchinfo/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,12 +3,12 @@
from .torchinfo import summary

__all__ = (
"summary",
"ColumnSettings",
"Mode",
"ModelStatistics",
"RowSettings",
"Units",
"Verbosity",
"summary",
)
__version__ = "1.8.0"
6 changes: 2 additions & 4 deletions torchinfo/formatting.py
Original file line number Diff line number Diff line change
Expand Up @@ -135,10 +135,8 @@ def layers_to_str(self, summary_list: list[LayerInfo], total_params: int) -> str
"""
new_str = ""
for layer_info in summary_list:
if (
layer_info.depth > self.max_depth
or self.hide_recursive_layers
and layer_info.is_recursive
if layer_info.depth > self.max_depth or (
self.hide_recursive_layers and layer_info.is_recursive
):
continue

Expand Down

0 comments on commit d937bbd

Please sign in to comment.