Fix positiveness of model randomizations

robotology-legacy · Oct 3, 2021 · dfb9129 · dfb9129
1 parent 7d74257
commit dfb9129
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python/gym_ignition/randomizers/model/sdf.py b/python/gym_ignition/randomizers/model/sdf.py
@@ -297,9 +297,6 @@ def sample(self, pretty_print=False) -> str:
             else:
                 raise ValueError("Distribution not recognized")
 
-            if data.force_positive:
-                sample = max(sample, 0.0)
-
             # Update the value
             if data.method is Method.Absolute:
 
@@ -318,6 +315,9 @@ def sample(self, pretty_print=False) -> str:
             else:
                 raise ValueError("Method not recognized")
 
+            if data.force_positive:
+                data.element.text = str(max(float(data.element.text), 0.0))
+
         return etree.tostring(self._root, pretty_print=pretty_print).decode()
 
     def new_randomization(self) -> RandomizationDataBuilder: