From 74f32e637862e2a37b77476b242ae1da7fb4033d Mon Sep 17 00:00:00 2001
From: 0xMochan <mochanbot@gmail.com>
Date: Thu, 17 Oct 2024 10:25:34 -0500
Subject: [PATCH] fix: adding cfg test

---
 rig-core/src/loaders/file.rs | 1 +
 rig-core/src/loaders/pdf.rs  | 1 +
 2 files changed, 2 insertions(+)

diff --git a/rig-core/src/loaders/file.rs b/rig-core/src/loaders/file.rs
index b8ff3d0c..eef36d61 100644
--- a/rig-core/src/loaders/file.rs
+++ b/rig-core/src/loaders/file.rs
@@ -100,6 +100,7 @@ impl<'a, State> FileLoader<'a, State> {
     }
 }
 
+#[cfg(test)]
 mod tests {
     use super::FileLoader;
 
diff --git a/rig-core/src/loaders/pdf.rs b/rig-core/src/loaders/pdf.rs
index 51fb4f9c..372bb0e9 100644
--- a/rig-core/src/loaders/pdf.rs
+++ b/rig-core/src/loaders/pdf.rs
@@ -144,6 +144,7 @@ impl<'a, State> PdfFileLoader<'a, State> {
     }
 }
 
+#[cfg(test)]
 mod tests {
     use super::PdfFileLoader;