From e758431cb3d25249d8b7c4a9fdbb9fe7adb67447 Mon Sep 17 00:00:00 2001
From: SilasMarvin <19626586+SilasMarvin@users.noreply.github.com>
Date: Mon, 18 Dec 2023 16:09:51 -0800
Subject: [PATCH] Added filename and modulename

---
 pgml-extension/src/bindings/mod.rs | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/pgml-extension/src/bindings/mod.rs b/pgml-extension/src/bindings/mod.rs
index 13702106d..79e543490 100644
--- a/pgml-extension/src/bindings/mod.rs
+++ b/pgml-extension/src/bindings/mod.rs
@@ -17,9 +17,11 @@ macro_rules! create_pymodule {
             pyo3::Python::with_gil(|py| -> anyhow::Result<pyo3::Py<pyo3::types::PyModule>> {
                 use $crate::bindings::TracebackError;
                 let src = include_str!(concat!(env!("CARGO_MANIFEST_DIR"), $pyfile));
-                Ok(pyo3::types::PyModule::from_code(py, src, "", "")
-                    .format_traceback(py)?
-                    .into())
+                Ok(
+                    pyo3::types::PyModule::from_code(py, src, "transformers.py", "__main__")
+                        .format_traceback(py)?
+                        .into(),
+                )
             })
         });
     };