Merge pull request #817 from PrefectHQ/sorting-hat

Add sorting hat example

docs/examples/hogwarts_sorting_hat/hogwarts_sorting_hat.md

@@ -0,0 +1,21 @@
+# Hogwarts sorting hat
+
+![](hogwarts_patch.webp){width="400"}
+
+
+!!! example "Hogwarts sorting hat"
+    ```python
+    import marvin
+
+    student = "Brave, daring, chivalrous, and sometimes a bit reckless."
+
+    house = marvin.classify(
+        student,
+        labels=["Gryffindor", "Hufflepuff", "Ravenclaw", "Slytherin"]
+    )
+    ```
+
+    !!! success "Welcome to Gryffindor!"
+        ```python
+        assert house == "Gryffindor"
+        ```

docs/examples/xkcd_bird.md

@@ -1,4 +1,4 @@
-# xkcd's bird photo classifier
+# xkcd bird classifier
 
 [![](https://imgs.xkcd.com/comics/tasks.png)](https://xkcd.com/1425/)
 

mkdocs.yml

@@ -82,6 +82,7 @@ nav:
     - Examples:
       - examples/xkcd_bird.md
       - examples/michael_scott_business/michael_scott_business.md
+      - examples/hogwarts_sorting_hat/hogwarts_sorting_hat.md
 
 
   - Community:

tests/ai/test_classify.py

@@ -71,3 +71,14 @@ class TestAsync:
         async def test_classify_positive_sentiment(self):
             result = await marvin.classify_async("This is a great feature!", bool)
             assert result is True
+
+    class TestExamples:
+        async def test_hogwarts_sorting_hat(self):
+            description = "Brave, daring, chivalrous, and sometimes a bit reckless."
+
+            house = marvin.classify(
+                description,
+                labels=["Gryffindor", "Hufflepuff", "Ravenclaw", "Slytherin"],
+            )
+
+            assert house == "Gryffindor"