From c2fe01093741455dd5b1678cc5a547f2a3a7447a Mon Sep 17 00:00:00 2001 From: "Tobenna P. Igwe" Date: Wed, 24 Nov 2021 04:55:02 +0900 Subject: [PATCH] Remove duplicate import of Haiku Module I believe the import of haiku is meant to be the second one within the `try` block as it is an optional dependency. PS: Sorry for yet another 1-line PR --- treex/nn/__init__.py | 1 - 1 file changed, 1 deletion(-) diff --git a/treex/nn/__init__.py b/treex/nn/__init__.py index 41218aaf..da9f416b 100644 --- a/treex/nn/__init__.py +++ b/treex/nn/__init__.py @@ -5,7 +5,6 @@ from .dropout import Dropout from .flatten import Flatten from .flax_module import FlaxModule -from .haiku_module import HaikuModule from .linear import Linear from .mlp import MLP from .sequential import Lambda, Sequential, sequence