early_start.py 656 B

12345678910111213141516
  1. from sqlalchemy.orm import Session
  2. from app.crud.base import CRUDBase
  3. from app.models.ml_models_path.early_start import EarlyStartDTRModelPath
  4. from app.schemas.model_path.early_start import EarlyStartDTRModelPathCreate, EarlyStartDTRModelPathUpdate
  5. class CRUDModelPathEarlyStartDTR(
  6. CRUDBase[EarlyStartDTRModelPath, EarlyStartDTRModelPathCreate, EarlyStartDTRModelPathUpdate]
  7. ):
  8. def get_path_by_device(self, db: Session, device_id: str) -> EarlyStartDTRModelPath:
  9. return db.query(self.model).filter(EarlyStartDTRModelPath.device_id == device_id).first()
  10. model_path_early_start_dtr = CRUDModelPathEarlyStartDTR(EarlyStartDTRModelPath)