Fix flake8 warnings (newlines, line wraps, spacing)

Author: arianatan

.DS_Store

@@ -0,0 +1,13 @@
+[flake8]
+max-line-length = 100
+extend-ignore = E203, W503
+exclude =
+    .git,
+    __pycache__,
+    build,
+    dist,
+    .venv,
+    venv,
+    fastapi,
+    env,
+    model/*.pkl

.flake8

@@ -1,16 +1,12 @@
-import json
-
 import requests
 
 # TODO: send a GET using the URL http://127.0.0.1:8000
-r = None # Your code here
+r = requests.get("http://127.0.0.1:8000")
 
 # TODO: print the status code
-# print()
+print("Status Code:", r.status_code)
 # TODO: print the welcome message
-# print()
-
-
+print("Result:", r.json().get("message"))
 
 data = {
     "age": 37,
@@ -30,9 +26,9 @@
 }
 
 # TODO: send a POST using the data above
-r = None # Your code here
+r = requests.post("http://127.0.0.1:8000/data/", json=data)
 
 # TODO: print the status code
-# print()
+print("Status Code:", r.status_code)
 # TODO: print the result
-# print()
+print("Result:", r.json().get("result"))

.idea/.gitignore

@@ -7,6 +7,7 @@
 from ml.data import apply_label, process_data
 from ml.model import inference, load_model
 
+
 # DO NOT MODIFY
 class Data(BaseModel):
     age: int = Field(..., example=37)
@@ -26,21 +27,24 @@ class Data(BaseModel):
     hours_per_week: int = Field(..., example=40, alias="hours-per-week")
     native_country: str = Field(..., example="United-States", alias="native-country")
 
-path = None # TODO: enter the path for the saved encoder 
+
+PROJECT_PATH = os.path.dirname(os.path.abspath(__file__))
+MODEL_DIR = os.path.join(PROJECT_PATH, "model")
+
+path = os.path.join(MODEL_DIR, "encoder.pkl")
 encoder = load_model(path)
 
-path = None # TODO: enter the path for the saved model 
+path = os.path.join(MODEL_DIR, "model.pkl")
 model = load_model(path)
 
-# TODO: create a RESTful API using FastAPI
-app = None # your code here
+app = FastAPI(title="Census Income Inference API", version="1.0.0")
+
 
 # TODO: create a GET on the root giving a welcome message
 @app.get("/")
 async def get_root():
     """ Say hello!"""
-    # your code here
-    pass
+    return {"message": "Hello from the API!"}
 
 
 # TODO: create a POST on a different path that does model inference
@@ -65,10 +69,15 @@ async def post_inference(data: Data):
         "native-country",
     ]
     data_processed, _, _, _ = process_data(
-        # your code here
+        data,
+        categorical_features=cat_features,
+        label=None,
+        training=False,
+        encoder=encoder,
+        lb=None,
         # use data as data input
         # use training = False
         # do not need to pass lb as input
     )
-    _inference = None # your code here to predict the result using data_processed
+    _inference = inference(model, data_processed)
     return {"result": apply_label(_inference)}