Skip to content

Commit cb5fecd

Browse files
committed
fix ruff error
1 parent 65ad267 commit cb5fecd

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

pandasai/data_loader/semantic_layer_schema.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,9 @@ class Column(BaseModel):
4747
name: str = Field(..., description="Name of the column.")
4848
type: Optional[str] = Field(None, description="Data type of the column.")
4949
description: Optional[str] = Field(None, description="Description of the column")
50-
expression: Optional[str] = Field(None, description="Aggregation expression (avg, min, max, sum)")
50+
expression: Optional[str] = Field(
51+
None, description="Aggregation expression (avg, min, max, sum)"
52+
)
5153
alias: Optional[str] = Field(None, description="Alias for the column")
5254

5355
@field_validator("type")
@@ -73,8 +75,12 @@ def is_expression_valid(cls, expr: str) -> Optional[str]:
7375

7476
class Relation(BaseModel):
7577
name: Optional[str] = Field(None, description="Name of the relationship.")
76-
description: Optional[str] = Field(None, description="Description of the relationship.")
77-
from_: str = Field(..., alias="from", description="Source column for the relationship.")
78+
description: Optional[str] = Field(
79+
None, description="Description of the relationship."
80+
)
81+
from_: str = Field(
82+
..., alias="from", description="Source column for the relationship."
83+
)
7884
to: str = Field(..., description="Target column for the relationship.")
7985

8086

0 commit comments

Comments
 (0)