summaryrefslogtreecommitdiffstats
path: root/mlscraper/training.py
blob: 875ba220f502ca787ee73c0037f7f35afdba20dd (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
# training objects
from mlscraper.parser import Page


class MultiItemPageSample:
    """Sample of an item on a page containing several items."""

    page = None
    items = None

    def __init__(self, page: Page, items: list):
        self.page = page
        self.items = items


class SingleItemPageSample:
    """Sample of an item on a page containing one item only."""

    page = None
    item = None

    def __init__(self, page: Page, item: dict):
        self.page = page
        self.item = item