Skip to content

Commit db25fa2

Browse files
fix(fix-basepath): basepath sync
1 parent 64e141f commit db25fa2

File tree

2 files changed

+9
-2
lines changed

2 files changed

+9
-2
lines changed

predictor/app.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -60,8 +60,8 @@ async def predict(
6060

6161
os.makedirs(base_path, exist_ok=True)
6262
meta_path, results_path = (
63-
os.path.join(output_path, "meta"),
64-
os.path.join(output_path, "results"),
63+
os.path.join(base_path, "meta"),
64+
os.path.join(base_path, "results"),
6565
)
6666
os.makedirs(meta_path, exist_ok=True)
6767
os.makedirs(results_path, exist_ok=True)

predictor/models.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -91,6 +91,13 @@ class PredictionRequest(BaseModel):
9191
default=False,
9292
description="Whether to include predictions as points, this will create output geojson with extra points predictions",
9393
)
94+
remove_metadata: Optional[bool] = Field(
95+
default=True,
96+
description="Whether to remove intermediate metadata files after processing",
97+
)
98+
output_path: Optional[str] = Field(
99+
default=None, description="Path to save the output files"
100+
)
94101

95102
@field_validator("checkpoint")
96103
def validate_checkpoint(cls, value):

0 commit comments

Comments
 (0)