Fix for type annotation

experimaestro · Dec 7, 2023 · 7e4a911 · 7e4a911
1 parent ee84874
commit 7e4a911
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/xpmir/letor/samplers/__init__.py b/src/xpmir/letor/samplers/__init__.py
@@ -358,7 +358,7 @@ def initialize(self, random):
         super().initialize(random)
         self.sampler.initialize(random)
 
-    def pairwise_iter(self) -> SerializableIterator[PairwiseRecord]:
+    def pairwise_iter(self) -> SerializableIterator[PairwiseRecord, Any]:
         def iter(pair_iter):
             while True:
                 topics = []