Try to resolve #2018

Signed-off-by: Simon Zhao <[email protected]>
recommenders-team · Jan 30, 2024 · 6f1eab5 · 6f1eab5
1 parent 2c6b258
commit 6f1eab5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/recommenders/models/deeprec/models/sequential/rnn_cell_implement.py b/recommenders/models/deeprec/models/sequential/rnn_cell_implement.py
@@ -581,9 +581,9 @@ def __init__(
     ):
         self._build_bias = build_bias
 
-        if args is None or (nest.is_sequence(args) and not args):
+        if args is None or (nest.is_nested(args) and not args):
             raise ValueError("`args` must be specified")
-        if not nest.is_sequence(args):
+        if not nest.is_nested(args):
             args = [args]
             self._is_sequence = False
         else:

diff --git a/setup.py b/setup.py
@@ -60,7 +60,7 @@
 extras_require = {
     "gpu": [
         "nvidia-ml-py3>=7.352.0",
-        "tensorflow==2.8.4",  # FIXME: Temporarily pinned due to issue with TF version > 2.10.1 See #2018
+        "tensorflow~=2.8.4,~=2.9.3,>=2.10.1",
         "tf-slim>=1.1.0",
         "torch>=1.13.1",  # for CUDA 11 support
         "fastai>=1.0.46,<2",