Path	Lines of Code
tests/__init__.py	1
tests/fixtures/__init__.py	1
tests/fixtures/adapter_config.py	7
tests/fixtures/loggers.py	6
tests/hyperpod_nemo_adapter/collections/data/datasets/test_dummy_dataset.py	22
tests/hyperpod_nemo_adapter/collections/data/datasets/test_dummy_dpo_dataset.py	36
tests/hyperpod_nemo_adapter/collections/data/datasets/test_hf_dataset.py	46
tests/hyperpod_nemo_adapter/collections/data/datasets/test_hf_dpo_dataset.py	46
tests/hyperpod_nemo_adapter/collections/data/test_base.py	62
tests/hyperpod_nemo_adapter/collections/data/test_dummy_data_module.py	22
tests/hyperpod_nemo_adapter/collections/data/test_dummy_dpo_data_module.py	50
tests/hyperpod_nemo_adapter/collections/data/test_hf_data_module.py	30
tests/hyperpod_nemo_adapter/collections/data/test_hf_dpo_data_module.py	63
tests/hyperpod_nemo_adapter/collections/model/test_sagemaker_base_model.py	305
tests/hyperpod_nemo_adapter/collections/parts/test_fsdp_strategy.py	62
tests/hyperpod_nemo_adapter/collections/parts/test_sagemaker_trainer_builder.py	21
tests/hyperpod_nemo_adapter/conf/test_config_schemas.py	422
tests/hyperpod_nemo_adapter/patches/test_patch_llama_flash_attn_cp.py	52
tests/hyperpod_nemo_adapter/utils/callbacks/e2e/test_callback_io_creation.py	58
tests/hyperpod_nemo_adapter/utils/callbacks/e2e/test_combined_ckpt.py	59
tests/hyperpod_nemo_adapter/utils/callbacks/e2e/test_data_module.py	76
tests/hyperpod_nemo_adapter/utils/callbacks/e2e/test_full_ckpt.py	110
tests/hyperpod_nemo_adapter/utils/callbacks/e2e/test_hybrid_shard_ckpt.py	149
tests/hyperpod_nemo_adapter/utils/callbacks/e2e/test_lr_scheduler.py	45
tests/hyperpod_nemo_adapter/utils/callbacks/e2e/test_peft_ckpt.py	152
tests/hyperpod_nemo_adapter/utils/callbacks/e2e/test_resilience_ckpt.py	263
tests/hyperpod_nemo_adapter/utils/callbacks/e2e/test_sharded_ckpt.py	116
tests/hyperpod_nemo_adapter/utils/callbacks/e2e/test_utils.py	296
tests/hyperpod_nemo_adapter/utils/callbacks/test_ckpt_io.py	74
tests/hyperpod_nemo_adapter/utils/callbacks/test_full_ckpt_io.py	36
tests/hyperpod_nemo_adapter/utils/callbacks/test_local_ckpt_io.py	32
tests/hyperpod_nemo_adapter/utils/callbacks/test_peft_ckpt_io.py	118
tests/hyperpod_nemo_adapter/utils/callbacks/test_sharded_ckpt_io.py	82
tests/hyperpod_nemo_adapter/utils/test_config_utils.py	64
tests/hyperpod_nemo_adapter/utils/test_exp_manager.py	62
tests/hyperpod_nemo_adapter/utils/test_fsdp_utils.py	66
tests/hyperpod_nemo_adapter/utils/test_sm_utils.py	83
tests/hyperpod_nemo_adapter/utils/test_tracer_utils.py	87
tests/hyperpod_nemo_adapter/utils/test_train_utils.py	65
tests/test_utils.py	21
tests/utils.py	9