workspace.fit add return_nll argument