Fix macro when exported

egraphs-good · Dec 2, 2024 · 555faf4 · 555faf4
1 parent a8b2aa1
commit 555faf4
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/sort/macros.rs b/src/sort/macros.rs
@@ -6,16 +6,16 @@ macro_rules! add_primitives {
         let type_info: &mut _ = $type_info;
         #[allow(unused_imports, non_snake_case)]
         {
-            use $crate::{*, sort::*, constraint::*};
+            use $crate::{*, ast::*, sort::*, constraint::*};
 
             struct MyPrim {$(
                 $param: Arc<<$param_t as FromSort>::Sort>,
             )*
                 __out: Arc<<$ret as IntoSort>::Sort>,
             }
 
-            impl $crate::PrimitiveLike for MyPrim {
-                fn name(&self) -> $crate::Symbol {
+            impl PrimitiveLike for MyPrim {
+                fn name(&self) -> Symbol {
                     $name.into()
                 }
 
@@ -45,7 +45,7 @@ macro_rules! add_primitives {
                     }
                 }
             }
-            type_info.add_primitive($crate::Primitive::from(MyPrim {
+            type_info.add_primitive( Primitive::from(MyPrim {
                 $( $param: type_info.get_sort_nofail::<<$param_t as IntoSort>::Sort>(), )*
                 __out: type_info.get_sort_nofail::<<$ret as IntoSort>::Sort>(),
             }))