diff --git a/code/aind_auto_train/schema/generate_diagrams.py b/code/aind_auto_train/schema/generate_diagrams.py
new file mode 100644
index 0000000..9aa06a1
--- /dev/null
+++ b/code/aind_auto_train/schema/generate_diagrams.py
@@ -0,0 +1,17 @@
+""" Generate schema diagrams using erdantic
+Install erdantic here https://erdantic.drivendata.org/v0.6/
+"""
+
+import erdantic as erd
+from aind_auto_train.schema.curriculum import Curriculum, DynamicForagingCurriculum, DummyTaskCurriculum
+
+erd.draw(DynamicForagingCurriculum,
+         out="./code/aind_auto_train/schema/schema_diagram_DynamicForagingCurriculum.png")
+
+erd.draw(DummyTaskCurriculum,
+         out="./code/aind_auto_train/schema/schema_diagram_DummyTaskCurriculum.png")
+
+erd.draw(Curriculum,
+         out="./code/aind_auto_train/schema/schema_diagram_Curriculum.png")
+
+
diff --git a/code/aind_auto_train/schema/schema_diagram_Curriculum.png b/code/aind_auto_train/schema/schema_diagram_Curriculum.png
new file mode 100644
index 0000000..a3bd282
Binary files /dev/null and b/code/aind_auto_train/schema/schema_diagram_Curriculum.png differ
diff --git a/code/aind_auto_train/schema/schema_diagram_DummyTaskCurriculum.png b/code/aind_auto_train/schema/schema_diagram_DummyTaskCurriculum.png
new file mode 100644
index 0000000..a3e240d
Binary files /dev/null and b/code/aind_auto_train/schema/schema_diagram_DummyTaskCurriculum.png differ
diff --git a/code/aind_auto_train/schema/schema_diagram_DynamicForagingCurriculum.png b/code/aind_auto_train/schema/schema_diagram_DynamicForagingCurriculum.png
new file mode 100644
index 0000000..d8ea6dc
Binary files /dev/null and b/code/aind_auto_train/schema/schema_diagram_DynamicForagingCurriculum.png differ