trying to fix DEM tests

KratosMultiphysics · Oct 17, 2018 · 75a78e1 · 75a78e1
1 parent d50fea7
commit 75a78e1
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/applications/DEM_application/python_scripts/DEM_procedures.py b/applications/DEM_application/python_scripts/DEM_procedures.py
@@ -1311,7 +1311,7 @@ def __init__(self, post_path, name, step, which_folder):
 
 class DEMIo(object):
 
-    def __init__(self, model, DEM_parameters, post_path):
+    def __init__(self, DEM_parameters, post_path, all_model_parts):
 
         self.post_path = post_path
         self.mixed_model_part = model.CreateModelPart("Mixed_Part")

diff --git a/applications/DEM_application/python_scripts/DEM_procedures_mpi.py b/applications/DEM_application/python_scripts/DEM_procedures_mpi.py
@@ -140,8 +140,8 @@ def __init__(self,name,step):
 
 class DEMIo(DEM_procedures.DEMIo):
 
-    def __init__(self, model, DEM_parameters, post_path):
-        super(DEMIo,self).__init__(DEM_parameters, post_path)
+    def __init__(self, DEM_parameters, post_path, all_model_parts):
+        super(DEMIo,self).__init__(DEM_parameters, post_path, all_model_parts)
 
     def AddMpiVariables(self):
         self.spheres_variables.append(PARTITION_INDEX)

diff --git a/applications/DEM_application/python_scripts/main_script.py b/applications/DEM_application/python_scripts/main_script.py
@@ -544,7 +544,7 @@ def CleanUpOperations(self):
             del self.DEM_inlet
 
     def SetGraphicalOutput(self):
-        self.demio = DEM_procedures.DEMIo(self.model, self.DEM_parameters, self.post_path)
+        self.demio = DEM_procedures.DEMIo(self.DEM_parameters, self.post_path, self.all_model_parts)
         if self.DEM_parameters["post_vtk_option"].GetBool():
             import dem_vtk_output
             self.vtk_output = dem_vtk_output.VtkOutput(self.main_path, self.problem_name, self.spheres_model_part, self.rigid_face_model_part)