From c6cfd8892b836e9e123c876faca085dacd78ee1a Mon Sep 17 00:00:00 2001
From: Rafael Ravedutti <rafaelravedutti@gmail.com>
Date: Fri, 29 Dec 2023 17:30:11 +0100
Subject: [PATCH] Small fixes

Signed-off-by: Rafael Ravedutti <rafaelravedutti@gmail.com>
---
 runtime/vtk.hpp       | 1 +
 src/pairs/sim/comm.py | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/runtime/vtk.hpp b/runtime/vtk.hpp
index 2fed101..f122565 100644
--- a/runtime/vtk.hpp
+++ b/runtime/vtk.hpp
@@ -31,6 +31,7 @@ void vtk_write_data(PairsSimulation *ps, const char *filename, int start, int en
 
     ps->copyPropertyToHost(masses, ReadOnly);
     ps->copyPropertyToHost(positions, ReadOnly);
+    ps->copyPropertyToHost(flags, ReadOnly);
 
     for(int i = start; i < end; i++) {
         if(flags(i) & FLAGS_INFINITE) {
diff --git a/src/pairs/sim/comm.py b/src/pairs/sim/comm.py
index 9462228..3a439d8 100644
--- a/src/pairs/sim/comm.py
+++ b/src/pairs/sim/comm.py
@@ -47,7 +47,7 @@ class Comm:
         # Every property that is not constant across timesteps and have neighbor accesses during any
         # interaction kernel (i.e. property[j] in force calculation kernel)
         prop_names = ['position', 'linear_velocity', 'angular_velocity']
-        prop_list = [ self.sim.property(p) for p in prop_names if self.sim.property(p) is not None]
+        prop_list = [self.sim.property(p) for p in prop_names if self.sim.property(p) is not None]
 
         PackAllGhostParticles(self, prop_list)
         CommunicateAllData(self, prop_list)
-- 
GitLab