diff --git a/pgml-sdks/pgml/src/lib.rs b/pgml-sdks/pgml/src/lib.rs
index 74a5c529b..cef33c024 100644
--- a/pgml-sdks/pgml/src/lib.rs
+++ b/pgml-sdks/pgml/src/lib.rs
@@ -13,6 +13,7 @@ use tracing::Level;
 use tracing_subscriber::FmtSubscriber;
 
 mod builtins;
+#[cfg(any(feature = "python", feature = "javascript"))]
 mod cli;
 mod collection;
 mod filter_builder;