Skip to content

Commit

Permalink
More linting fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
wsfulton committed Nov 20, 2024
1 parent f97c38e commit fafadcd
Showing 1 changed file with 5 additions and 4 deletions.
9 changes: 5 additions & 4 deletions abi3audit/_state.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,15 +6,16 @@

import os
import sys
from typing import Any, Literal, Optional, Type
from types import TracebackType
from typing import Any, Literal

from rich.console import Console
from rich.status import Status


class StatusWrapper:
_status: Status | None

def __init__(self, console: Console) -> None:
self._console = console
self._status = None
Expand All @@ -26,9 +27,9 @@ def __enter__(self) -> StatusWrapper:

def __exit__(
self,
exc_type: Optional[Type[BaseException]],
exc_val: Optional[BaseException],
exc_tb: Optional[TracebackType],
exc_type: type[BaseException] | None,
exc_val: BaseException | None,
exc_tb: TracebackType | None,
) -> Any:
if self._status:
return self._status.__exit__(exc_type, exc_val, exc_tb)
Expand Down

0 comments on commit fafadcd

Please sign in to comment.