Skip to content

Commit

Permalink
auto fixes from pre-commit hooks
Browse files Browse the repository at this point in the history
  • Loading branch information
pre-commit-ci[bot] committed Sep 2, 2024
1 parent e67c526 commit a03783a
Show file tree
Hide file tree
Showing 10 changed files with 16 additions and 8 deletions.
3 changes: 2 additions & 1 deletion tests/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@
from unittest.mock import PropertyMock, patch

import pytest

from revnets.context import context as context_
from revnets.context.context import Context
from revnets.models import Config, Evaluation, HyperParameters, Path
Expand All @@ -22,7 +23,7 @@ def _mocked_assets_path() -> Iterator[None]:
yield


@pytest.fixture()
@pytest.fixture
def test_context(
context: Context,
_mocked_assets_path: None, # noqa: PT019
Expand Down
6 changes: 3 additions & 3 deletions tests/evaluations/test_evaluations.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,14 +4,14 @@
from typing import cast

import pytest
from torch.nn import Sequential

from revnets import evaluations, pipelines, reconstructions
from revnets.context import Context
from revnets.evaluations import analysis, attack, outputs, weights
from revnets.evaluations.evaluate import format_percentage
from revnets.networks import mininet
from revnets.pipelines import Pipeline
from torch.nn import Sequential

from tests.evaluations import verifier
from tests.evaluations.verifier import Standardization

Expand Down Expand Up @@ -40,7 +40,7 @@
)


@pytest.fixture()
@pytest.fixture
def pipeline() -> Pipeline:
return pipelines.mininet.Pipeline()

Expand Down
2 changes: 1 addition & 1 deletion tests/evaluations/test_standardize.py
Original file line number Diff line number Diff line change
@@ -1,9 +1,9 @@
from types import ModuleType

import pytest

from revnets import networks
from revnets.models import Activation

from tests.evaluations.verifier import Standardization, Verifier

standardizations = (
Expand Down
3 changes: 2 additions & 1 deletion tests/evaluations/test_utils.py
Original file line number Diff line number Diff line change
@@ -1,13 +1,14 @@
from typing import cast

from hypothesis import given, strategies
from torch import nn

from revnets.standardization.utils import (
extract_layer,
extract_rnn_layers,
extract_scale_isomorphism_after_max_pool,
)
from revnets.standardization.weights import rnn
from torch import nn


@given(
Expand Down
3 changes: 2 additions & 1 deletion tests/evaluations/verifier.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,8 @@
from typing import cast

import torch
from torch.nn import Module, Sequential

from revnets import standardization
from revnets.context import context
from revnets.models import Activation
Expand All @@ -18,7 +20,6 @@
extract_internal_layers,
)
from revnets.standardization.weights import feedforward
from torch.nn import Module, Sequential


class Standardization(Enum):
Expand Down
1 change: 1 addition & 0 deletions tests/test_cli_entry_points.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
from unittest.mock import MagicMock, patch

from package_dev_utils.tests.args import no_cli_args

from revnets.cli import entry_point, launch


Expand Down
1 change: 1 addition & 0 deletions tests/test_data.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@
from typing import TYPE_CHECKING

import pytest

from revnets import data

if TYPE_CHECKING:
Expand Down
1 change: 1 addition & 0 deletions tests/test_main.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
import pytest
import torch

from revnets.main.main import Experiment

gpu_available = torch.cuda.is_available()
Expand Down
1 change: 1 addition & 0 deletions tests/test_pipelines.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@
from typing import TYPE_CHECKING

import pytest

from revnets import pipelines

if TYPE_CHECKING:
Expand Down
3 changes: 2 additions & 1 deletion tests/test_reconstructions.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,12 +2,13 @@

import pytest
from pytorch_lightning.core.optimizer import LightningOptimizer
from torch import nn

from revnets import reconstructions
from revnets.pipelines.mininet import Pipeline
from revnets.reconstructions import Reconstructor, queries
from revnets.training.reconstructions import Network
from revnets.training.reconstructions.callbacks import LearningRateScheduler
from torch import nn

reconstruction_modules = (
reconstructions.empty,
Expand Down

0 comments on commit a03783a

Please sign in to comment.