Some personal fixes to the CMU pytensorlib