Fix compatibility

janheinrichmerker · Oct 14, 2024 · a793bf6 · a793bf6
1 parent d13da48
commit a793bf6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ray_elasticsearch/__init__.py b/ray_elasticsearch/__init__.py
@@ -87,7 +87,7 @@ def num_rows(self) -> int:
         return self._num_rows
 
     @cached_property
-    def _estimated_inmemory_data_size(self) -> int | None:
+    def _estimated_inmemory_data_size(self) -> Optional[int]:
         stats = self._elasticsearch.indices.stats(
             index=self._index,
             metric="store",
@@ -96,7 +96,7 @@ def _estimated_inmemory_data_size(self) -> int | None:
             return None
         return stats["_all"]["total"]["store"]["total_data_set_size_in_bytes"]
 
-    def estimate_inmemory_data_size(self) -> int | None:
+    def estimate_inmemory_data_size(self) -> Optional[int]:
         return self._estimated_inmemory_data_size
 
     @staticmethod