#482: clean up

DARMA-tasking · Dec 14, 2023 · 3ee28b7 · 3ee28b7
1 parent db8cfdb
commit 3ee28b7
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 4 deletions.
diff --git a/src/lbaf/IO/lbsVTDataReader.py b/src/lbaf/IO/lbsVTDataReader.py
@@ -2,9 +2,8 @@
 import os
 import re
 from logging import Logger
-from multiprocessing import get_context
+from multiprocessing import get_context, Manager
 from multiprocessing.pool import Pool
-from multiprocessing import Manager
 
 import brotli
 

diff --git a/src/lbaf/IO/lbsVTDataWriter.py b/src/lbaf/IO/lbsVTDataWriter.py
@@ -84,7 +84,7 @@ def __create_tasks(self, rank_id, objects, migratable):
 
         return tasks
 
-    def __get_communications(self, phase, rank): # CWS: in progress
+    def __get_communications(self, phase, rank):
         """Create communication entries to be outputted to JSON."""
 
         # Get initial communications (if any)

diff --git a/tests/unit/config/conf_vt_writer_communications_test.yaml b/tests/unit/config/conf_vt_writer_communications_test.yaml
@@ -31,7 +31,7 @@ algorithm:
 output_dir: ../output/vt_writer_communications_test
 output_file_stem: output_file
 write_JSON:
-  compressed: false
+  compressed: true
   suffix: json
   communications: true
   offline_LB_compatible: false