ruff errors

pyterrier_adaptive/__init__.py

@@ -3,3 +3,5 @@
 from .gar import GAR
 from .corpus_graph import CorpusGraph, NpTopKCorpusGraph
 from pyterrier_adaptive._laff import Laff
+
+__all__ = ['GAR', 'CorpusGraph', 'NpTopKCorpusGraph', 'Laff']

pyterrier_adaptive/corpus_graph.py

@@ -1,17 +1,14 @@
 import pickle
 import tempfile
 from lz4.frame import LZ4FrameFile
-import shutil
 import json
 import numpy as np
 import pandas as pd
 from pathlib import Path
-import torch
 import more_itertools
 from typing import Union, Tuple, List
 import ir_datasets
 from npids import Lookup
-import pyterrier as pt
 import pyterrier_alpha as pta
 
 try:

pyterrier_adaptive/gar.py

@@ -4,6 +4,7 @@
 import pyterrier as pt
 import pandas as pd
 import ir_datasets
+import pyterrier_adaptive
 logger = ir_datasets.log.easy()
 
 
@@ -20,7 +21,7 @@ class GAR(pt.Transformer):
     """
     def __init__(self,
         scorer: pt.Transformer,
-        corpus_graph: 'CorpusGraph',
+        corpus_graph: 'pyterrier_adaptive.CorpusGraph',
         num_results: int = 1000,
         batch_size: Optional[int] = None,
         backfill: bool = True,

requirements-dev.txt

@@ -1,3 +1,4 @@
+ruff
 pytest
 pytest-subtests
 pytest-cov