Spaces:

bethgelab
/

lm-similarity

Sleeping

Joschka Strueber commited on Feb 7

Commit

7c4f6b6

1 Parent(s): 001064a

[Fix] error in filter_responses call

Files changed (1) hide show

src/dataloading.py CHANGED Viewed

@@ -161,7 +161,7 @@ def load_run_data(model_name, dataset_name):
         data = data.to_dict()
         # Get ground truth labels and logits
-        log_probs = filter_responses(dataset_name, data)
         labels = filter_labels(dataset_name, data["doc"])
     except Exception as e:

         data = data.to_dict()
         # Get ground truth labels and logits
+        log_probs = filter_responses(data)
         labels = filter_labels(dataset_name, data["doc"])
     except Exception as e: