Tests for augmentations

[KGallyamov]

Pull Request Template for InnoFW

Title: Augmentations test (also minor changes in datasets and schedulers tests)

Description

Wrote unit tests coverage for augmentations, augmentations_train/test/val *.yaml configs.

Pre-commit Hook Activation and Linter Checks

Please ensure you have activated the pre-commit hook to enable automatic linting checks before submitting your PR. This helps maintain code quality and consistency throughout the project.

Documentation

Please ensure you have documented the functionality of your changes, especially if it introduces new features or affects existing ones. Proper documentation ensures that the project remains maintainable and understandable for other contributors.

Changes

• List major changes made in bullet points
• Include any minor changes or refactoring

What type of PR is this? (check all applicable)

• Refactor
• Feature
• Bug Fix
• Optimization
• Documentation Update

Checklist

• I have tested my changes locally and all tests passed
• I have updated the documentation (if necessary)
• I have added new dependencies (if necessary)
• My changes do not introduce any breaking changes or conflicts with existing code
• I have activated the pre-commit hook and ensured that linting checks pass

Added/updated tests?

• Yes
• No, and this is why: please replace this line with details on why tests
  have not been included
• I need help with writing tests

Additional Notes

Provide any additional information or context that may be helpful for the reviewer.

---

Thank you for your contribution to InnoFW! Your efforts help make this project better for everyone.