Re-raise stancsv parsing failures to identify file

Author: amas0

cmdstanpy/stanfit/gq.py

@@ -626,10 +626,16 @@ def _assemble_generated_quantities(self) -> None:
             order='F',
         )
         for chain in range(self.chains):
-            *_, draws = stancsv.parse_comments_header_and_draws(
-                self.runset.csv_files[chain]
-            )
-            gq_sample[:, chain, :] = stancsv.csv_bytes_list_to_numpy(draws)
+            csv_file = self.runset.csv_files[chain]
+            try:
+                *_, draws = stancsv.parse_comments_header_and_draws(
+                    self.runset.csv_files[chain]
+                )
+                gq_sample[:, chain, :] = stancsv.csv_bytes_list_to_numpy(draws)
+            except Exception as exc:
+                raise ValueError(
+                    f"An error occurred when parsing Stan csv {csv_file}"
+                ) from exc
         self._draws = gq_sample
 
     def _draws_start(self, inc_warmup: bool) -> Tuple[int, int]:

cmdstanpy/stanfit/laplace.py

@@ -86,10 +86,16 @@ def _assemble_draws(self) -> None:
         if self._draws.shape != (0,):
             return
 
-        *_, draws = stancsv.parse_comments_header_and_draws(
-            self._runset.csv_files[0]
-        )
-        self._draws = stancsv.csv_bytes_list_to_numpy(draws)
+        csv_file = self._runset.csv_files[0]
+        try:
+            *_, draws = stancsv.parse_comments_header_and_draws(
+                self._runset.csv_files[0]
+            )
+            self._draws = stancsv.csv_bytes_list_to_numpy(draws)
+        except Exception as exc:
+            raise ValueError(
+                f"An error occurred when parsing Stan csv {csv_file}"
+            ) from exc
 
     def stan_variable(self, var: str) -> np.ndarray:
         """

cmdstanpy/stanfit/metadata.py

@@ -35,8 +35,15 @@ def __init__(
     def from_csv(
         cls, stan_csv: Union[str, os.PathLike, Iterator[bytes]]
     ) -> 'InferenceMetadata':
-        comments, header, _ = stancsv.parse_comments_header_and_draws(stan_csv)
-        return cls(stancsv.construct_config_header_dict(comments, header))
+        try:
+            comments, header, _ = stancsv.parse_comments_header_and_draws(
+                stan_csv
+            )
+            return cls(stancsv.construct_config_header_dict(comments, header))
+        except Exception as exc:
+            raise ValueError(
+                f"An error occurred when parsing Stan csv {stan_csv}"
+            ) from exc
 
     def __repr__(self) -> str:
         return 'Metadata:\n{}\n'.format(self._cmdstan_config)

cmdstanpy/stanfit/mle.py

@@ -35,15 +35,24 @@ def __init__(self, runset: RunSet) -> None:
         )  # make the typechecker happy
         self._save_iterations: bool = optimize_args.save_iterations
 
-        (
-            comment_lines,
-            header,
-            draws_lines,
-        ) = stancsv.parse_comments_header_and_draws(self.runset.csv_files[0])
-        self._metadata = InferenceMetadata(
-            stancsv.construct_config_header_dict(comment_lines, header)
-        )
-        all_draws = stancsv.csv_bytes_list_to_numpy(draws_lines)
+        csv_file = self.runset.csv_files[0]
+        try:
+            (
+                comment_lines,
+                header,
+                draws_lines,
+            ) = stancsv.parse_comments_header_and_draws(
+                self.runset.csv_files[0]
+            )
+            self._metadata = InferenceMetadata(
+                stancsv.construct_config_header_dict(comment_lines, header)
+            )
+            all_draws = stancsv.csv_bytes_list_to_numpy(draws_lines)
+
+        except Exception as exc:
+            raise ValueError(
+                f"An error occurred when parsing Stan csv {csv_file}"
+            ) from exc
         self._mle: np.ndarray = all_draws[-1]
         if self._save_iterations:
             self._all_iters: np.ndarray = all_draws

cmdstanpy/stanfit/pathfinder.py

@@ -78,10 +78,16 @@ def _assemble_draws(self) -> None:
         if self._draws.shape != (0,):
             return
 
-        *_, draws = stancsv.parse_comments_header_and_draws(
-            self._runset.csv_files[0]
-        )
-        self._draws = stancsv.csv_bytes_list_to_numpy(draws)
+        csv_file = self._runset.csv_files[0]
+        try:
+            *_, draws = stancsv.parse_comments_header_and_draws(
+                self._runset.csv_files[0]
+            )
+            self._draws = stancsv.csv_bytes_list_to_numpy(draws)
+        except Exception as exc:
+            raise ValueError(
+                f"An error occurred when parsing Stan csv {csv_file}"
+            ) from exc
 
     def stan_variable(self, var: str) -> np.ndarray:
         """

cmdstanpy/stanfit/vb.py

@@ -29,18 +29,27 @@ def __init__(self, runset: RunSet) -> None:
             )
         self.runset = runset
 
-        (
-            comment_lines,
-            header,
-            draw_lines,
-        ) = stancsv.parse_comments_header_and_draws(self.runset.csv_files[0])
-
-        self._metadata = InferenceMetadata(
-            stancsv.construct_config_header_dict(comment_lines, header)
-        )
-        self._eta = stancsv.parse_variational_eta(comment_lines)
+        csv_file = self.runset.csv_files[0]
+        try:
+            (
+                comment_lines,
+                header,
+                draw_lines,
+            ) = stancsv.parse_comments_header_and_draws(
+                self.runset.csv_files[0]
+            )
 
-        draws_np = stancsv.csv_bytes_list_to_numpy(draw_lines)
+            self._metadata = InferenceMetadata(
+                stancsv.construct_config_header_dict(comment_lines, header)
+            )
+            self._eta = stancsv.parse_variational_eta(comment_lines)
+
+            draws_np = stancsv.csv_bytes_list_to_numpy(draw_lines)
+
+        except Exception as exc:
+            raise ValueError(
+                f"An error occurred when parsing Stan csv {csv_file}"
+            ) from exc
         self._variational_mean: np.ndarray = draws_np[0]
         self._variational_sample: np.ndarray = draws_np[1:]