From 9aefc9896ce09036702d6caf913f6f7ee0edd1fa Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?C=C3=A9dric=20Deltheil?= Date: Thu, 8 Feb 2024 18:35:08 +0000 Subject: [PATCH] test_trainer: use `model_copy` instead of `copy` The `copy` method has been deprecated. --- tests/training_utils/test_trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/training_utils/test_trainer.py b/tests/training_utils/test_trainer.py index 55e769f..6e8a493 100644 --- a/tests/training_utils/test_trainer.py +++ b/tests/training_utils/test_trainer.py @@ -82,7 +82,7 @@ def mock_trainer(mock_config: MockConfig) -> MockTrainer: @pytest.fixture def mock_trainer_short(mock_config: MockConfig) -> MockTrainer: - mock_config_short = mock_config.copy() + mock_config_short = mock_config.model_copy(deep=True) mock_config_short.training.duration = {"number": 3, "unit": TimeUnit.STEP} return MockTrainer(config=mock_config_short)