remove experimental features

ASEM000 · Jul 15, 2023 · 3fcf63e · 3fcf63e
1 parent 88b79f7
commit 3fcf63e
Show file tree

Hide file tree

Showing 4 changed files with 0 additions and 256 deletions.
diff --git a/serket/experimental/__init__.py b/serket/experimental/__init__.py
@@ -11,7 +11,3 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-
-from .lazy_class import lazy_class
-
-__all__ = ["lazy_class"]
diff --git a/serket/experimental/lazy_class.py b/serket/experimental/lazy_class.py
diff --git a/serket/experimental/test_lazy_class.py b/serket/experimental/test_lazy_class.py
diff --git a/serket/nn/recurrent.py b/serket/nn/recurrent.py
@@ -977,7 +977,6 @@ def __init__(
         backward_cell: RNNCell | None = None,
         *,
         return_sequences: bool = False,
-        return_state: bool = False,
     ):
         if not isinstance(cell, RNNCell):
             raise TypeError(f"Expected {cell=} to be an instance of RNNCell.")