Reddit / models /types.py
cyrilfrl's picture
Upload 14 files
690dc6b verified
# standard-library imports
import typing as tt
# third-party imports
import torch
# types definitions
Title = str
Text = str
Subreddit = str
Fetch_UTC = float
Hours_Ago = float
Count = int
Tensor = torch.Tensor
RawSample = tt.Tuple[Title, Text, Subreddit, Fetch_UTC, Hours_Ago]
ProcessedSample = tt.Tuple[Tensor, Tensor, int, float, Hours_Ago,
Count, Count, Count]