diff options
Diffstat (limited to 'mlscraper/training.py')
| -rw-r--r-- | mlscraper/training.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/mlscraper/training.py b/mlscraper/training.py index 875ba22..28517a4 100644 --- a/mlscraper/training.py +++ b/mlscraper/training.py @@ -22,3 +22,7 @@ class SingleItemPageSample: def __init__(self, page: Page, item: dict): self.page = page self.item = item + + def find_nodes(self, attr): + needle = self.item[attr] + return self.page.find(needle) |