Skip to content

Commit 568d1f8

Browse files
emlinfacebook-github-bot
authored andcommitted
forward fix compatibility test regression (#3179)
Summary: Pull Request resolved: #3179 this is to fix regression test caused by new parameter virtual_table_eviction_policy in embedding bag Reviewed By: PoojaAg18 Differential Revision: D78099867 fbshipit-source-id: 9e5faa4a40b1ff3908eb0f8778b618698b18800e
1 parent 0a5233d commit 568d1f8

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

torchrec/schema/api_tests/test_embedding_config_schema.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818
EmbeddingBagConfig,
1919
EmbeddingConfig,
2020
PoolingType,
21+
VirtualTableEvictionPolicy,
2122
)
2223

2324
from torchrec.schema.utils import is_signature_compatible
@@ -41,6 +42,7 @@ class StableEmbeddingBagConfig:
4142
input_dim: Optional[int] = None
4243
total_num_buckets: Optional[int] = None
4344
use_virtual_table: bool = False
45+
virtual_table_eviction_policy: Optional[VirtualTableEvictionPolicy] = None
4446
pooling: PoolingType = PoolingType.SUM
4547

4648

0 commit comments

Comments
 (0)