import datetime

from collections.abc import Sequence

from typing import overload

import numpy as np

import numpy.typing as npt

TZ = str | datetime.tzinfo

def _get_tzinfo(tz: TZ | None=None) -> datetime.tzinfo: ...

def _reset_epoch_test_example() -> None: ...

def set_epoch(epoch: str) -> None: ...

def get_epoch() -> str: ...

def _dt64_to_ordinalf(d: npt.NDArray[np.datetime64]) -> npt.NDArray[np.floating]: ...

def _from_ordinalf(x: float, tz: TZ | None=None) -> datetime.datetime: ...

# Ideally str | Sequence[str] would get an override, but because a str is a valid Sequence[str],

# it's not possible to distinguish between them in the type system

# See https://github.com/python/typing/issues/256

def datestr2num(d: str | Sequence[str], default: datetime.datetime | None=None) -> float | npt.NDArray[np.floating]: ...

@overload

def date2num(d: datetime.datetime | np.datetime64) -> float: ...

@overload

def date2num(d: Sequence[datetime.datetime] | Sequence[np.datetime64]) -> npt.NDArray[np.floating]: ...

@overload

def num2date(x: float, tz: TZ | None=None) -> datetime.datetime: ...

@overload

def num2date(x: Sequence[float], tz: TZ | None=None) -> list[datetime.datetime]: ...

@overload

def num2timedelta(x: float) -> datetime.timedelta: ...

@overload

def num2timedelta(x: Sequence[float]) -> list[datetime.timedelta]: ...

def drange(dstart: datetime.datetime, dend: datetime.datetime, delta: datetime.timedelta) -> npt.NDArray[np.floating]: ...

def _wrap_in_tex(text: str) -> str: ...