Skip to content

Commit

Permalink
adding validate() in estimator
Browse files Browse the repository at this point in the history
  • Loading branch information
hangqianjun committed Sep 18, 2024
1 parent 14d1772 commit 6b36c8d
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 1 deletion.
2 changes: 1 addition & 1 deletion src/rail/core/data.py
Original file line number Diff line number Diff line change
Expand Up @@ -244,7 +244,7 @@ def make_name(cls, tag):

@classmethod
def _check_data_columns(cls, path, columns_to_check, parent_groupname=None, **kwargs):
raise NotImplementedError
raise NotImplementedError # pragma: no cover


class TableHandle(DataHandle):
Expand Down
1 change: 1 addition & 0 deletions src/rail/estimation/estimator.py
Original file line number Diff line number Diff line change
Expand Up @@ -100,6 +100,7 @@ def estimate(self, input_data):
Handle providing access to QP ensemble with output data
"""
self.set_data("input", input_data)
self.validate()
self.run()
self.finalize()
return self.get_handle("output")
Expand Down

0 comments on commit 6b36c8d

Please sign in to comment.