| Total Complexity | 1 |
| Total Lines | 19 |
| Duplicated Lines | 0 % |
| Changes | 0 | ||
| 1 | """Validation checks for scikit-learn integration.""" |
||
| 2 | |||
| 3 | |||
| 4 | class Checks: |
||
| 5 | """Checks class.""" |
||
| 6 | |||
| 7 | _fit_successful = False |
||
| 8 | |||
| 9 | def verify_fit(function): |
||
| 10 | """Verify Fit function.""" |
||
| 11 | |||
| 12 | def wrapper(self, X, y): |
||
| 13 | """Wrap function call.""" |
||
| 14 | out = function(self, X, y) |
||
| 15 | self._fit_successful = True |
||
| 16 | return out |
||
| 17 | |||
| 18 | return wrapper |
||
| 19 |