From c2f41eb39210b93752826aa956c77bc57d53c73b Mon Sep 17 00:00:00 2001 From: LordBaryhobal Date: Tue, 7 Jul 2026 13:43:57 +0200 Subject: [PATCH] fix: make Frame/Column inherit from pandas classes --- midas/typing.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/midas/typing.py b/midas/typing.py index fc121ce..1496c95 100644 --- a/midas/typing.py +++ b/midas/typing.py @@ -1,6 +1,8 @@ from typing import Generic, TypeVar from typing import cast as typing_cast +from pandas import DataFrame, Series + cast = typing_cast """### Midas documentation Cast a value to a type. @@ -38,14 +40,14 @@ _**Internal Python documentation**_ T = TypeVar("T") -class Frame(Generic[T]): +class Frame(DataFrame, Generic[T]): """A `Frame` is the abstract type implemented by `DataFrame` A frame contains any number of named columns (see :class:`Column`) """ -class Column(Generic[T]): +class Column(Series, Generic[T]): """A `Column` is the abstract type implemented by `Series` A column contains a any number of values of the same type