Coverage for  / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / trainer / config / __init__.py: 100%

1# Copyright 2026 Huawei Technologies Co., Ltd 

2# 

3# Licensed under the Apache License, Version 2.0 (the "License"); 

4# you may not use this file except in compliance with the License. 

5# You may obtain a copy of the License at 

6# 

7# http://www.apache.org/licenses/LICENSE-2.0 

8# 

9# Unless required by applicable law or agreed to in writing, software 

10# distributed under the License is distributed on an "AS IS" BASIS, 

11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 

12# See the License for the specific language governing permissions and 

13# limitations under the License. 

14# ============================================================================ 

15"""Typed trainer configuration tree (stage 7, 05 §15.2.5). 

16 

17The former ``auto_models/trainer/config.py`` monolith is split into 

18``target`` / ``training`` / ``parallelism`` / ``optimization`` / ``data`` / 

19``trainer`` by configuration domain; ``manager`` and ``resolver`` moved 

20from ``auto_models/config`` unchanged. This package re-exports the same 

21public class names — including the AutoModels-owned ``CompileConfig`` / 

22``FSDP2Config`` / ``CheckpointingConfig`` — without copying their 

23definitions. 

24""" 

25 

26__all__ = [ 

27 "AcceleratorConfig", 

28 "ActivationCheckpointConfig", 

29 "ActivationCheckpointSelection", 

30 "CompileConfig", 

31 "DataLoaderConfig", 

32 "DatasetConfig", 

33 "DebugConfig", 

34 "FSDP2Config", 

35 "MemoryConfig", 

36 "MixedPrecisionConfig", 

37 "ModelIntegrationConfig", 

38 "OptimizerConfig", 

39 "OptimizerSwapConfig", 

40 "ProfilingAnalysisConfig", 

41 "ProfilingConfig", 

42 "Target", 

43 "TrainerConfig", 

44 "TrainingConfig", 

45 "WandbConfig", 

46 "save_configs", 

47 "CheckpointingConfig", 

48] 

49 

50from hyper_parallel.models.build_options import CompileConfig, FSDP2Config 

51from hyper_parallel.components.checkpoint.config import CheckpointingConfig 

52 

53from hyper_parallel.trainer.config.data import ( 

54 DataLoaderConfig, 

55 DatasetConfig, 

56 ModelAssetsConfig, 

57) 

58from hyper_parallel.trainer.config.optimization import ( 

59 MixedPrecisionConfig, 

60 OptimizerConfig, 

61 OptimizerSwapConfig, 

62) 

63from hyper_parallel.trainer.config.parallelism import ( 

64 AcceleratorConfig, 

65 ActivationCheckpointConfig, 

66 ActivationCheckpointSelection, 

67 PlanOverride, 

68 _import_module_type, 

69 entries_to_module_replacements, 

70 entries_to_plan_overrides, 

71 normalize_distributed_setup_overrides, 

72) 

73from hyper_parallel.trainer.config.target import Target 

74from hyper_parallel.trainer.config.trainer import TrainerConfig, save_configs 

75from hyper_parallel.trainer.config.training import ( 

76 DebugConfig, 

77 MemoryConfig, 

78 ModelIntegrationConfig, 

79 ProfilingAnalysisConfig, 

80 ProfilingConfig, 

81 TrainingConfig, 

82 WandbConfig, 

83)