models/types.py

# 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]