Commit 756eb251 authored by Johan Gudmundsson's avatar Johan Gudmundsson
Browse files

clean up warning msg

parent b46e28b0
Loading
Loading
Loading
Loading
+2 −2
Original line number Diff line number Diff line
@@ -71,9 +71,9 @@ def borch_proxy_class(
    """
    if cls in BORCHIFY_REGISTRY:
        return BORCHIFY_REGISTRY[cls]
    if issubclass(cls, nn.Sequential) and cls is not nn.Sequential:
    if issubclass(cls, nn.Sequential):
        warnings.warn(
            f"Skipping Borchifying {cls.__name__} as it is a subclass of `Sequential`",
            f"Skipping Borchifying `{cls.__name__}` as it is a subclass of `Sequential`",
            UserWarning,
        )
        return cls