from typing import Optional, Callable, Tuple, Dict, List, Union
import torch
import numpy as np

Tensor = torch.Tensor
NumpyArray = np.ndarray

__all__ = ["Optional", "Callable", "Tuple", "Dict", "List", "Tensor", "Union"]
