Loading src/borch/nn/borchify.py +2 −2 Original line number Diff line number Diff line Loading @@ -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 Loading Loading
src/borch/nn/borchify.py +2 −2 Original line number Diff line number Diff line Loading @@ -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 Loading