SIENTIAPDE-1321: Updated tests

tests/utils/models/test_train_model_result.py

@@ -175,11 +175,11 @@ def test_train_model_result_is_dataclass(sample_params, sample_dataframes):
 
 
 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
 
     result_fields = fields(TrainModelResult)
-    assert len(result_fields) == 17
+    assert len(result_fields) == 19
 
     field_names = {f.name for f in result_fields}
     expected_fields = {
@@ -195,6 +195,8 @@ def test_train_model_result_field_count():
         'mse_val',
         'mae_val',
         'r2_val',
+        'equation',
+        'equation_path',
         'run_name',
         'report_path',
         'train_data_path',