lint

mepland · Nov 11, 2023 · 37ba684 · 37ba684
1 parent de382f6
commit 37ba684
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/utils/TSModelWrappers.py b/utils/TSModelWrappers.py
@@ -996,7 +996,9 @@ def __init__(self: "NBEATSModelWrapper", **kwargs: Any) -> None:  # noqa: ANN401
         # but they do not work if the kwargs["TSModelWrapper"] parent instance was updated between child __init__ calls
         if (
             "TSModelWrapper" in kwargs
-            and type(kwargs["TSModelWrapper"].__class__)  # pylint: disable=unidiomatic-typecheck
+            and type(  # noqa: E721 # pylint: disable=unidiomatic-typecheck
+                kwargs["TSModelWrapper"].__class__
+            )
             == type(TSModelWrapper)  # <class 'type'>
             and str(kwargs["TSModelWrapper"].__class__)
             == str(TSModelWrapper)  # <class 'utils.TSModelWrappers.TSModelWrapper'>