Info that gradients could not agree

lab-cosmo · Oct 27, 2023 · c858bed · c858bed
1 parent 63c9c54
commit c858bed
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/equisolve/numpy/models/linear_model.py b/src/equisolve/numpy/models/linear_model.py
@@ -51,7 +51,7 @@ def _validate_data(self, X: TensorMap, y: Optional[TensorMap] = None) -> None:
             X, y, check=["samples", "components"]
         ):
             raise ValueError(
-                "Metadata (samples, components) of X and y does not agree!"
+                "Metadata (samples, components, gradients) of X and y does not agree!"
             )
 
     def _validate_params(
@@ -69,7 +69,8 @@ def _validate_params(
         """
         if not metatensor.equal_metadata(X, alpha, check=["components", "properties"]):
             raise ValueError(
-                "Metadata (components, properties) of X and alpha does not agree!"
+                "Metadata (components, properties, gradients) of X and alpha does "
+                "not agree!"
             )
 
         if sample_weight is not None and not metatensor.equal_metadata(
@@ -81,7 +82,8 @@ def _validate_params(
             ],
         ):
             raise ValueError(
-                "Metadata (samples, components) of X and sample_weight does not agree!"
+                "Metadata (samples, components, gradients) of X and sample_weight "
+                "does not agree!"
             )
 
         for key, alpha_block in alpha.items():