fix(types): add str methods to dataframe types
This commit is contained in:
@@ -160,11 +160,18 @@ class ConstraintType:
|
|||||||
class ColumnType:
|
class ColumnType:
|
||||||
type: Type
|
type: Type
|
||||||
|
|
||||||
|
def __str__(self) -> str:
|
||||||
|
return f"Column[{self.type}]"
|
||||||
|
|
||||||
|
|
||||||
@dataclass(frozen=True, kw_only=True)
|
@dataclass(frozen=True, kw_only=True)
|
||||||
class DataFrameType:
|
class DataFrameType:
|
||||||
columns: list[Column]
|
columns: list[Column]
|
||||||
|
|
||||||
|
def __str__(self) -> str:
|
||||||
|
schema: list[str] = [f"{col.name}: {col.type}" for col in self.columns]
|
||||||
|
return f"Frame[{', '.join(schema)}]"
|
||||||
|
|
||||||
@dataclass(frozen=True, kw_only=True)
|
@dataclass(frozen=True, kw_only=True)
|
||||||
class Column:
|
class Column:
|
||||||
index: int
|
index: int
|
||||||
|
|||||||
Reference in New Issue
Block a user