fix: downgrade sientia_do version and update CSV saving logic

- Downgraded `sientia_do` version to 1.11.0 in `requirements.txt` to address compatibility issues.
- Updated `data_manager_repository.py` to save training and test data CSVs using float-cast versions of the data, ensuring consistency in the saved outputs.
- Enhanced test coverage in `test_data_manager_repository.py` to verify the presence of the target alias in the generated CSV files.
This commit is contained in:
vitor-aignosi
2026-05-05 16:26:47 -03:00
parent 23912d854e
commit 78167f7a44
3 changed files with 15 additions and 5 deletions

View File

@@ -494,7 +494,7 @@ def test_generate_report_adds_target_alias_for_reports(tmp_path):
):
instance = mrep.return_value
instance.save_all_sections_html = Mock()
repo.generate_report(tmr, {})
out = repo.generate_report(tmr, {})
kwargs = mrep.call_args.kwargs
reference_data = kwargs['reference_data']
@@ -504,6 +504,15 @@ def test_generate_report_adds_target_alias_for_reports(tmp_path):
assert reference_data['target'].equals(reference_data['t'])
assert current_data['target'].equals(current_data['t'])
assert out.train_data_path is not None
assert out.test_data_path is not None
train_csv = pd.read_csv(out.train_data_path)
test_csv = pd.read_csv(out.test_data_path)
assert 'target' in train_csv.columns
assert 'target' in test_csv.columns
assert train_csv['target'].equals(train_csv['t'])
assert test_csv['target'].equals(test_csv['t'])
def test_generate_report_skips_equation_file_when_not_linear(tmp_path):
repo = dmr.DataManagerRepository(MagicMock())