diff --git a/pyproject.toml b/pyproject.toml
index e2ecd0d..814d8d1 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -2,4 +2,5 @@
 profile = "black"
 known_bytelatent = "bytelatent"
 known_apps = "apps"
+known_third_party = "wandb"
 sections = "FUTURE,STDLIB,THIRDPARTY,BYTELATENT,APPS,FIRSTPARTY,LOCALFOLDER"