Black

capreolus-ir · Aug 22, 2021 · afe19f1 · afe19f1
1 parent f5e5fc4
commit afe19f1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/capreolus/extractor/bertpassage.py b/capreolus/extractor/bertpassage.py
@@ -330,7 +330,7 @@ def _prepare_bert_input(self, query_toks, psg_toks):
         return inp, mask, seg
 
     def id2vec_for_pairs(self, qid, posid, negid=None, label=None, reldocs=None):
-        return self.id2vec_for_triplets(qid, posid,negid=negid, label=label)
+        return self.id2vec_for_triplets(qid, posid, negid=negid, label=label)
 
     def id2vec_for_triplets(self, qid, posid, negid=None, label=None):
         """

diff --git a/capreolus/tests/test_collection.py b/capreolus/tests/test_collection.py
@@ -10,6 +10,7 @@
 # TODO: Allow the below collections too to be tested
 collections = collections - {"gov2passages", "robust04passages"}
 
+
 @pytest.mark.parametrize("collection_name", collections)
 def test_collection_creatable(tmpdir_as_cache, collection_name):
     collection = Collection.create(collection_name)