diff --git a/pymc/model.py b/pymc/model.py
index 90311da95f2..067e6454cfe 100644
--- a/pymc/model.py
+++ b/pymc/model.py
@@ -1155,7 +1155,7 @@ def set_initval(self, rv_var, initval):
     def set_data(
         self,
         name: str,
-        values: Dict[str, Optional[Sequence]],
+        values: Union[Sequence, np.ndarray],
         coords: Optional[Dict[str, Sequence]] = None,
     ):
         """Changes the values of a data variable in the model.