Skip to content

Commit

Permalink
fix docstring style
Browse files Browse the repository at this point in the history
  • Loading branch information
ziw-liu committed Oct 11, 2024
1 parent 2f2a38e commit 246d2e5
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions viscy/cli.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,9 +37,7 @@ def add_arguments_to_parser(self, parser) -> None:


def _setup_environment() -> None:
"""
Set log level and TF32 precision.
"""
"""Set log level and TF32 precision."""
log_level = os.getenv("VISCY_LOG_LEVEL", logging.INFO)
logging.getLogger("lightning.pytorch").setLevel(log_level)
torch.set_float32_matmul_precision("high")
Expand Down

0 comments on commit 246d2e5

Please sign in to comment.