Merge pull request #171 from muupan/fix-ci-error-mypy-prioritized

Fix mypy error due to numbers.Number
pfnet · Mar 14, 2022 · a721ddf · a721ddf
2 parents cee0777 + a733679
commit a721ddf
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/pfrl/collections/prioritized.py b/pfrl/collections/prioritized.py
@@ -1,5 +1,4 @@
 import collections
-from numbers import Number
 from typing import (
     Any,
     Callable,
@@ -243,7 +242,7 @@ def popleft(self) -> Optional[V]:
         return ret
 
 
-def _find(index_left: int, index_right: int, node: Node, pos: Number) -> int:
+def _find(index_left: int, index_right: int, node: Node, pos: float) -> int:
     if index_right - index_left == 1:
         return index_left
     else: