👕 noqa F403s

ENSTA-U2IS-AI · Oct 30, 2023 · 1e81d8f · 1e81d8f
1 parent ec7e844
commit 1e81d8f
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 5 deletions.
diff --git a/tests/test_optimization_procedures.py b/tests/test_optimization_procedures.py
@@ -4,8 +4,10 @@
 from torch_uncertainty.models.resnet import resnet18, resnet50
 from torch_uncertainty.models.vgg import vgg16
 from torch_uncertainty.models.wideresnet import wideresnet28x10
-from torch_uncertainty.optimization_procedures import *
-from torch_uncertainty.optimization_procedures import get_procedure
+from torch_uncertainty.optimization_procedures import (
+    get_procedure,
+    optim_regression,
+)
 
 
 class TestOptProcedures:

diff --git a/torch_uncertainty/models/resnet/__init__.py b/torch_uncertainty/models/resnet/__init__.py
@@ -1,4 +1,4 @@
-# ruff: noqa: F401
+# ruff: noqa: F401, F403
 from .batched import *
 from .masked import *
 from .mimo import *

diff --git a/torch_uncertainty/models/vgg/__init__.py b/torch_uncertainty/models/vgg/__init__.py
@@ -1,3 +1,3 @@
-# ruff: noqa: F401
+# ruff: noqa: F401, F403
 from .packed import *
 from .std import *
diff --git a/torch_uncertainty/models/wideresnet/__init__.py b/torch_uncertainty/models/wideresnet/__init__.py
@@ -1,4 +1,4 @@
-# ruff: noqa: F401
+# ruff: noqa: F401, F403
 from .batched import *
 from .masked import *
 from .mimo import *