use max integer instead of float inf

usc-isi-i2 · Apr 16, 2018 · 51ab246 · 51ab246
1 parent fd20231
commit 51ab246
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/rltk/io/writer/block_file_writer.py b/rltk/io/writer/block_file_writer.py
@@ -1,11 +1,12 @@
 import json
 import os
+import sys
 
 from rltk.io.writer import BlockWriter
 
 
 class BlockFileWriter(BlockWriter):
-    def __init__(self, filename, buffer_size=10000, set_size=float('inf'), index_blacklist:set=None):
+    def __init__(self, filename, buffer_size=10000, set_size=sys.maxsize, index_blacklist:set=None):
         self._filename = filename
         self._temp_filename = filename + '.temp'
         self._buffer_size = buffer_size