diff --git a/README.md b/README.md
index ff11997..8367552 100644
--- a/README.md
+++ b/README.md
@@ -151,7 +151,7 @@ This is helpful when you want the ID of the prediction separate from its output.
 
 ```python
 prediction = replicate.predictions.create(
-    model="meta/meta-llama-3-70b-instruct"
+    model="meta/meta-llama-3-70b-instruct",
     input={"prompt": "Please write a haiku about llamas."},
     stream=True,
 )
diff --git a/pyproject.toml b/pyproject.toml
index 975f7d9..4e2a0d9 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "replicate"
-version = "1.0.4"
+version = "1.0.6"
 description = "Python client for Replicate"
 readme = "README.md"
 license = { file = "LICENSE" }
diff --git a/replicate/py.typed b/replicate/py.typed
new file mode 100644
index 0000000..e69de29
diff --git a/tests/test_client.py b/tests/test_client.py
index 1345a62..2c585b1 100644
--- a/tests/test_client.py
+++ b/tests/test_client.py
@@ -95,9 +95,9 @@ def test_custom_headers_are_applied():
 
     def mock_send(request):
         assert "User-Agent" in request.headers, "Custom header not found in request"
-        assert (
-            request.headers["User-Agent"] == "my-custom-user-agent/1.0"
-        ), "Custom header value is incorrect"
+        assert request.headers["User-Agent"] == "my-custom-user-agent/1.0", (
+            "Custom header value is incorrect"
+        )
         return httpx.Response(401, json={})
 
     mock_send_wrapper = mock.Mock(side_effect=mock_send)