diff --git a/website/docs/usage/projects.md b/website/docs/usage/projects.md
index b6688cd5d..04edf45bd 100644
--- a/website/docs/usage/projects.md
+++ b/website/docs/usage/projects.md
@@ -831,6 +831,8 @@ so you can compare their results. The following script starts an
 argument you can pass in from your `config.yml`: a comma-separated list of paths
 to load the pipelines from and an example text to use as the default text.
 
+<!-- TODO: replace with embed -->
+
 ```python
 ### scripts/visualize.py
 import spacy_streamlit
diff --git a/website/docs/usage/training.md b/website/docs/usage/training.md
index 9c18e4606..3f0141d72 100644
--- a/website/docs/usage/training.md
+++ b/website/docs/usage/training.md
@@ -683,7 +683,7 @@ You can also implement your own batch size schedule to use during training. The
 import spacy
 
 @spacy.registry.schedules("my_custom_schedule.v1")
-def my_custom_schedule(start: int = 1, factor: int = 1.001):
+def my_custom_schedule(start: int = 1, factor: float = 1.001):
    while True:
       yield start
       start = start * factor