Path	Lines of Code
test_installation.py	28
test_installation_main.py	108
torchrec/datasets/experimental/tests/__init__.py	1
torchrec/datasets/experimental/tests/test_criteo_dataframes.py	56
torchrec/datasets/scripts/tests/test_preprocess_criteo.py	29
torchrec/datasets/tests/__init__.py	1
torchrec/datasets/tests/criteo_test_utils.py	116
torchrec/datasets/tests/test_criteo.py	216
torchrec/datasets/tests/test_movielens.py	107
torchrec/datasets/tests/test_utils.py	106
torchrec/distributed/planner/tests/test_enumerators.py	526
torchrec/distributed/planner/tests/test_partitioners.py	349
torchrec/distributed/planner/tests/test_planners.py	78
torchrec/distributed/planner/tests/test_proposers.py	239
torchrec/distributed/planner/tests/test_shard_estimators.py	84
torchrec/distributed/tests/collective_utils_test.py	131
torchrec/distributed/tests/test_awaitable.py	22
torchrec/distributed/tests/test_comm.py	162
torchrec/distributed/tests/test_dist_data.py	371
torchrec/distributed/tests/test_fused_optim.py	261
torchrec/distributed/tests/test_lazy_awaitable.py	168
torchrec/distributed/tests/test_model.py	327
torchrec/distributed/tests/test_model_parallel.py	707
torchrec/distributed/tests/test_model_parallel_base.py	289
torchrec/distributed/tests/test_quant_model_parallel.py	150
torchrec/distributed/tests/test_train_pipeline.py	231
torchrec/distributed/tests/test_utils.py	250
torchrec/fx/tests/test_tracer.py	227
torchrec/linter/tests/test_module_linter.py	176
torchrec/models/tests/test_deepfm.py	161
torchrec/models/tests/test_dlrm.py	457
torchrec/modules/tests/__init__.py	1
torchrec/modules/tests/test_activation.py	18
torchrec/modules/tests/test_code_quality.py	23
torchrec/modules/tests/test_crossnet.py	125
torchrec/modules/tests/test_deepfm.py	130
torchrec/modules/tests/test_embedding_modules.py	206
torchrec/modules/tests/test_lazy_extension.py	198
torchrec/modules/tests/test_mlp.py	86
torchrec/modules/tests/test_score_learning.py	35
torchrec/optim/tests/__init__.py	1
torchrec/optim/tests/test_clipping.py	128
torchrec/optim/tests/test_keyed.py	189
torchrec/optim/tests/test_utils.py	7
torchrec/optim/tests/test_warmup.py	55
torchrec/quant/tests/test_embedding_modules.py	54
torchrec/sparse/tests/__init__.py	1
torchrec/sparse/tests/test_jagged_tensor.py	708
torchrec/sparse/tests/tests_utils.py	34
torchrec/tests/__init__.py	1
torchrec/tests/utils.py	80