Initial commit

rp_poetry/__init__.py

@@ -1 +1,2 @@
+__doc__ = "A simple package to generate poetry from a given text."
 __version__ = "0.1.0"

rp_poetry/main.py

@@ -7,6 +7,8 @@
 from fastapi import FastAPI
 import uvicorn
 
+fake_items_db = [{"item_name": "Foo"}, {"item_name": "Bar"}, {"item_name": "Baz"}]
+
 class ModelName(str, Enum):
     """
     Enumeration class for specifying model names.
@@ -97,12 +99,10 @@ async def get_model(model_name: ModelName):
     """
     if model_name == ModelName.ALEXNET:
         return {"model_name": model_name, "message": "Deep Learning FTW!"}
-    elif model_name == ModelName.LENET:
+    if model_name == ModelName.LENET:
         return {"model_name": model_name, "message": "LeCNN all the images"}
-    else:
-        return {"model_name": model_name, "message": "Have some residuals"}
 
-fake_items_db = [{"item_name": "Foo"}, {"item_name": "Bar"}, {"item_name": "Baz"}]
+    return {"model_name": model_name, "message": "Have some residuals"}
 
 @app.get("/items/")
 async def read_items(skip: int = 0, limit: int = 10):

tests/test_rp_poetry.py

@@ -1,5 +1,11 @@
+"""
+FastAPI Test application to serve models.
+"""
 from rp_poetry import __version__
 
 
 def test_version():
+    """
+    Test the version of the package.
+    """
     assert __version__ == "0.1.0"