From 42bf4bfad44d5aa3d23ac182adc6ad0afdd153e7 Mon Sep 17 00:00:00 2001
From: Aymeric Chaumont <aymeric.chaumont@student-cs.fr>
Date: Wed, 13 Jul 2022 14:16:11 +0200
Subject: [PATCH] linting

---
 backend/main.py          |  4 ++--
 backend/video-capture.py | 10 +++++-----
 2 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/backend/main.py b/backend/main.py
index cddefa0..7ab2713 100644
--- a/backend/main.py
+++ b/backend/main.py
@@ -47,7 +47,7 @@ async def get_records(place: str, db: Session = Depends(get_db)):
 
 
 @app.post('/api/records', response_model=schemas.Record)
-async def stats(record: schemas.RecordBase, db: Session = Depends(get_db)):
+async def post_records(record: schemas.RecordBase, db: Session = Depends(get_db)):
     db_record = models.Records(**record.dict())
     db.add(db_record)
     db.commit()
@@ -56,7 +56,7 @@ async def stats(record: schemas.RecordBase, db: Session = Depends(get_db)):
 
 
 @app.delete('/api/records', response_model=None)
-async def stats(id: int, db: Session = Depends(get_db)):
+async def del_records(id: int, db: Session = Depends(get_db)):
     if id == 0:
         db.query(models.Records).delete()
     else:
diff --git a/backend/video-capture.py b/backend/video-capture.py
index 9437c1b..530f4e6 100644
--- a/backend/video-capture.py
+++ b/backend/video-capture.py
@@ -3,7 +3,7 @@ from datetime import datetime, timedelta
 import numpy as np
 import keras
 from utils.preprocessing import fix_singular_shape, norm_by_imagenet
-from db import models, database, schemas
+from db import models
 
 from db.database import SessionLocal
 
@@ -24,13 +24,13 @@ while(cap.isOpened()):
         count_prediction = np.sum(pred_map)
         waiting_time = timedelta(seconds=120 + count_prediction * 30)
         record = {"place": "local",
-            "date": current_time,
-            "density": count_prediction,
-            "waiting_time": waiting_time}
+                  "date": current_time,
+                  "density": count_prediction,
+                  "waiting_time": waiting_time}
         db_record = models.Records(**record)
         db.add(db_record)
         db.commit()
         db.refresh(db_record)
     count += 1
 cap.release()
-cv2.destroyAllWindows()
\ No newline at end of file
+cv2.destroyAllWindows()
-- 
GitLab