mirror of
https://github.com/azaion/ai-training.git
synced 2026-04-22 07:06:36 +00:00
Refactor constants management to use Pydantic BaseModel for configuration
- Replaced module-level path variables in constants.py with a structured Pydantic Config class. - Updated all relevant modules (train.py, augmentation.py, exports.py, dataset-visualiser.py, manual_run.py) to access paths through the new config structure. - Fixed bugs related to image processing and model saving. - Enhanced test infrastructure to accommodate the new configuration approach. This refactor improves code maintainability and clarity by centralizing configuration management.
This commit is contained in:
@@ -4,3 +4,4 @@ markers =
|
||||
resilience: Resilience/error handling tests
|
||||
security: Security tests
|
||||
resource_limit: Resource limit tests
|
||||
e2e: End-to-end training pipeline tests (slow, require GPU/MPS)
|
||||
|
||||
Reference in New Issue
Block a user