Fix function import in compile/builders.py

Author: jessegrabowski

pytensor/compile/builders.py

@@ -6,7 +6,7 @@
 from typing import Optional, cast
 
 import pytensor.tensor as pt
-from pytensor.compile import function
+from pytensor.compile.function import function
 from pytensor.compile.function.pfunc import rebuild_collect_shared
 from pytensor.compile.mode import optdb
 from pytensor.compile.sharedvalue import SharedVariable

tests/tensor/rewriting/test_linalg.py

@@ -383,4 +383,4 @@ def test_local_lift_through_linalg(constructor, f_op, f, g_op, g):
 
     f2(*test_vals)
 
-    np.testing.assert_allclose(f1(*test_vals), f2(*test_vals))
+    np.testing.assert_allclose(f1(*test_vals), f2(*test_vals), atol=1e-8)