Improve pre-commit for mypy and black

TylerYep · Sep 9, 2023 · d34de25 · d34de25
1 parent 238e33d
commit d34de25
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -18,7 +18,7 @@ repos:
     hooks:
       - id: isort
 
-  - repo: https://github.com/psf/black
+  - repo: https://github.com/psf/black-pre-commit-mirror
     rev: 23.7.0
     hooks:
       - id: black

diff --git a/tests/torchinfo_xl_test.py b/tests/torchinfo_xl_test.py
@@ -52,15 +52,15 @@ def test_eval_order_doesnt_matter() -> None:
     )
     model1.eval()
     summary(model1, input_size=input_size)
-    with torch.inference_mode():  # type: ignore[no-untyped-call]
+    with torch.inference_mode():
         output1 = model1(input_tensor)
 
     model2 = torchvision.models.resnet18(
         weights=torchvision.models.ResNet18_Weights.DEFAULT
     )
     summary(model2, input_size=input_size)
     model2.eval()
-    with torch.inference_mode():  # type: ignore[no-untyped-call]
+    with torch.inference_mode():
         output2 = model2(input_tensor)
 
     assert torch.all(torch.eq(output1, output2))