[demo] Added seed for the demo

Tigul · Nov 29, 2024 · 416c2ef · 416c2ef
1 parent bc76292
commit 416c2ef
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/demos/pycram_bullet_world_demo/demo.py b/demos/pycram_bullet_world_demo/demo.py
@@ -10,7 +10,9 @@
 from pycram.datastructures.dataclasses import Color
 from pycram.ros.viz_marker_publisher import VizMarkerPublisher
 from pycrap import Robot, Apartment, Milk, Cereal, Spoon, Bowl
+import numpy as np
 
+np.random.seed(420)
 extension = ObjectDescription.get_file_extension()
 
 world = BulletWorld(WorldMode.GUI)

diff --git a/demos/pycram_bullet_world_demo/test_demo.py b/demos/pycram_bullet_world_demo/test_demo.py
@@ -3,6 +3,5 @@
 try:
     import demo
 except Exception as e:
-    print(e)
     World.current_world.exit()
-    exit(1)
+    raise e