Add checks against time coordinate if it was set by user before calling build_statespace_graph

Author: jessegrabowski

pymc_experimental/statespace/utils/data_tools.py

@@ -114,6 +114,16 @@ def add_data_to_active_model(values, index):
 
     if TIME_DIM not in pymc_mod.coords:
         pymc_mod.add_coord(TIME_DIM, index, mutable=True)
+    else:
+        found_time = pymc_mod.coords[TIME_DIM]
+        if found_time is None:
+            pymc_mod.coords.update({TIME_DIM: index})
+        elif not np.array_equal(found_time, index):
+            raise ValueError(
+                "Provided data has a different time index than the model. Please ensure that the time values "
+                "set on coords matches that of the exogenous data."
+            )
+
     data = pm.ConstantData("data", values, dims=data_dims)
 
     return data