Skip to content

Commit 9d2585e

Browse files
MoFHekarhdong
authored andcommitted
[fix] TypeVar was used incorrectly and Callable was used instead.
1 parent 9149973 commit 9d2585e

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tensorflow_recommenders_addons/dynamic_embedding/python/ops/dynamic_embedding_variable.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -190,7 +190,8 @@ def _insert_de_shard_from_file_system(
190190
value_dim,
191191
shard_keys_file_list,
192192
shard_values_file_list,
193-
partition_fn: typing.TypeVar(default_partition_fn) = default_partition_fn,
193+
partition_fn: typing.Callable[[typing.Any, typing.Any],
194+
typing.Any] = default_partition_fn,
194195
proc_size: int = None,
195196
proc_rank: int = None,
196197
buffer_size: int = 4194304):

0 commit comments

Comments
 (0)