Fix int typecast for rotatemaxfilesizebytes (#19)

Co-authored-by: Vikas-kum <[email protected]>
awslabs · Aug 5, 2020 · 257f31d · 257f31d
1 parent 480578f
commit 257f31d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/smdebug/profiler/profiler_config_parser.py b/smdebug/profiler/profiler_config_parser.py
@@ -95,7 +95,9 @@ def load_config(self):
 
         try:
             local_path = config.get("localpath", BASE_FOLDER_DEFAULT)
-            file_max_size = int(config.get("rotatemaxfilesizeinbytes", MAX_FILE_SIZE_DEFAULT))
+            file_max_size = int(
+                float(config.get("rotatemaxfilesizeinbytes", MAX_FILE_SIZE_DEFAULT))
+            )
             file_close_interval = float(
                 config.get("rotatefilecloseintervalinseconds", CLOSE_FILE_INTERVAL_DEFAULT)
             )