Clean up file & code

app/api/router/predict.py

@@ -30,11 +30,6 @@ def get_db():
 )
 
 
-@router.get("/")
-def hello_world():
-    return {"message": "Hello predict"}
-
-
 @router.put("/insurance")
 def predict_insurance(info: RegModelPrediction, db: Session = Depends(get_db)):
     """

app/api/schemas.py

@@ -1,50 +1,6 @@
 from pydantic import BaseModel
 
 
-class ItemBase(BaseModel):
-    title: str
-
-
-class ItemCreate(ItemBase):
-    pass
-
-
-class Item(ItemBase):
-    class Config:
-        orm_mode = True
-
-
-class DatasetBase(BaseModel):
-    path: str
-    version: int
-
-
-class DatasetCreate(DatasetBase):
-    pass
-
-
-class Dataset(DatasetBase):
-    class config:
-        orm_mode = True
-
-
-class ClfModelBase(BaseModel):
-    path: str
-    version: int
-    name: str
-    classes: int
-    score: float
-
-
-class ClfModelCreate(ClfModelBase):
-    pass
-
-
-class ClfModel(ClfModelBase):
-    class Config:
-        orm_mode = True
-
-
 class RegModelBase(BaseModel):
     model_name: str
 

app/crud.py

@@ -4,35 +4,6 @@
 from app.api import schemas
 
 
-def get_dataset(db: Session, version=1):
-    return db.query(models.Dataset).filter(
-        models.Dataset.version == version
-    ).first()
-
-
-def create_dataset(db: Session, Dataset: schemas.DatasetCreate):
-    db_dataset = models.Dataset(**Dataset)
-    db.add(db_dataset)
-    db.commit()
-    db.refresh(db_dataset)
-    return db_dataset
-
-
-def get_clf_model(db: Session, version=1, name='random_forest'):
-    return db.query(models.ClfModel).filter(
-        models.ClfModel.version == version and
-        models.ClfModel.name == name
-    ).first()
-
-
-def create_clf_model(db: Session, clf_model: schemas.ClfModelCreate):
-    db_cf_model = models.ClfModel(**clf_model)
-    db.add(db_cf_model)
-    db.commit()
-    db.refresh(db_cf_model)
-    return db_cf_model
-
-
 def get_reg_model(db: Session, model_name: schemas.RegModelBase):
     return db.query(models.RegModel).filter(
         models.RegModel.model_name == model_name

app/models.py

@@ -1,6 +1,6 @@
 # -*- coding: utf-8 -*-
 import datetime
-from sqlalchemy import Column, Integer, String, FLOAT, DateTime, ForeignKey
+from sqlalchemy import Column, String, FLOAT, DateTime, ForeignKey
 from sqlalchemy.sql.functions import now
 from sqlalchemy.orm import relationship
 
@@ -9,32 +9,6 @@
 KST = datetime.timezone(datetime.timedelta(hours=9))
 
 
-class Item(Base):
-    __tablename__ = 'items'
-
-    id = Column(Integer, primary_key=True, index=True, autoincrement=True)
-    title = Column(String, index=True, default='test')
-
-
-class Dataset(Base):
-    __tablename__ = 'dataset'
-
-    id = Column(Integer, primary_key=True, index=True, autoincrement=True)
-    path = Column(String, index=True)
-    version = Column(Integer, index=True, autoincrement=True)
-
-
-class ClfModel(Base):
-    __tablename__ = 'clf_model'
-
-    id = Column(Integer, primary_key=True, index=True, autoincrement=True)
-    path = Column(String, index=True)
-    version = Column(Integer, index=True, autoincrement=True)
-    name = Column(String, index=True)
-    classes = Column(Integer)
-    score = Column(FLOAT)
-
-
 class RegModel(Base):
     __tablename__ = 'reg_model'
 

main.py

@@ -1,8 +1,7 @@
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 
-from app.api.router import test, predict, train, upload
-from app.database import SessionLocal
+from app.api.router import predict
 
 app = FastAPI()
 
@@ -16,18 +15,7 @@
     allow_headers=["*"],
 )
 
-app.include_router(test.router)
 app.include_router(predict.router)
-app.include_router(upload.router)
-app.include_router(train.router)
-
-
-def get_db():
-    db = SessionLocal()
-    try:
-        yield db
-    finally:
-        db.close()
 
 
 @app.get("/")