Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-09 23:12:02 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

test_trainer: use model_copy instead of copy

Browse source 
The `copy` method has been deprecated.
This commit is contained in:
Cédric Deltheil 2024-02-08 18:35:08 +00:00 committed by Cédric Deltheil
parent f4aa0271b8
commit 9aefc9896c
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tests/training_utils/test_trainer.py
View file

@ -82,7 +82,7 @@ def mock_trainer(mock_config: MockConfig) -> MockTrainer:
@pytest.fixture @pytest.fixture
def mock_trainer_short(mock_config: MockConfig) -> MockTrainer: 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} mock_config_short.training.duration = {"number": 3, "unit": TimeUnit.STEP}
return MockTrainer(config=mock_config_short) return MockTrainer(config=mock_config_short)