Fix generating pytest for training service
This commit is contained in:
@@ -61,5 +61,10 @@ class TrainingSettings(BaseServiceSettings):
|
||||
# Distributed Training (for future scaling)
|
||||
DISTRIBUTED_TRAINING_ENABLED: bool = os.getenv("DISTRIBUTED_TRAINING_ENABLED", "false").lower() == "true"
|
||||
TRAINING_WORKER_COUNT: int = int(os.getenv("TRAINING_WORKER_COUNT", "1"))
|
||||
|
||||
PROPHET_DAILY_SEASONALITY: bool = True
|
||||
PROPHET_WEEKLY_SEASONALITY: bool = True
|
||||
PROPHET_YEARLY_SEASONALITY: bool = True
|
||||
PROPHET_SEASONALITY_MODE: str = "additive"
|
||||
|
||||
settings = TrainingSettings()
|
||||
Reference in New Issue
Block a user