Skip to content

Commit 61d278c

Browse files
committed
Save jaxpr in right path
1 parent 69e476a commit 61d278c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/lcm/solve_brute.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -127,7 +127,7 @@ def solve_continuous_problem(
127127

128128
with nvtx.annotate("jitting", color="white"):
129129
if period == 20:
130-
with open('/home/mj/Desktop/Jaxprs/function.txt', 'w') as f:
130+
with open('function.txt', 'w') as f:
131131
f.write(str(jax.make_jaxpr(_gridmapped)(
132132
**state_choice_space.dense_vars,
133133
**continuous_choice_grids,

0 commit comments

Comments
 (0)