Merge pull request #11 from Kogelvis/master

Set the solver when barnesHut isn't chosen
WestHealth · May 23, 2018 · b8a16bd · b8a16bd
2 parents 109c46d + c6ec1af
commit b8a16bd
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pyvis/physics.py b/pyvis/physics.py
@@ -97,12 +97,15 @@ def use_barnes_hut(self, params):
 
     def use_force_atlas_2based(self, params):
         self.forceAtlas2Based = self.forceAtlas2Based(params)
+        self.solver = 'forceAtlas2Based'
 
     def use_repulsion(self, params):
         self.repulsion = self.Repulsion(params)
+        self.solver = 'repulsion'
 
     def use_hrepulsion(self, params):
         self.hierarchicalRepulsion = self.hierarchicalRepulsion(params)
+        self.solver = 'hierarchicalRepulsion'
 
     def toggle_stabilization(self, status):
         self.stabilization.toggle_stabilization(status)