gitea_admin/sientia-dataops-model-manager
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SIENTIAPDE-1321: Updated tests

Browse Source
This commit is contained in:
Kou-Kinoshita
2025-10-27 19:45:47 -03:00
parent a80d65d7ba
commit 734221e32d
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/utils/models/test_train_model_result.py
View File

@@ -175,11 +175,11 @@ def test_train_model_result_is_dataclass(sample_params, sample_dataframes):
def test_train_model_result_field_count(): def test_train_model_result_field_count():
"""Test that TrainModelResult has exactly 17 fields.""" """Test that TrainModelResult has exactly 19 fields."""
from dataclasses import fields from dataclasses import fields
result_fields = fields(TrainModelResult) result_fields = fields(TrainModelResult)
assert len(result_fields) == 17 assert len(result_fields) == 19
field_names = {f.name for f in result_fields} field_names = {f.name for f in result_fields}
expected_fields = { expected_fields = {
@@ -195,6 +195,8 @@ def test_train_model_result_field_count():
'mse_val', 'mse_val',
'mae_val', 'mae_val',
'r2_val', 'r2_val',
'equation',
'equation_path',
'run_name', 'run_name',
'report_path', 'report_path',
'train_data_path', 'train_data_path',