Rectify type-hints for set_data

Author: Dhruvanshu-Joshi

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.